Skip to content

Instantly share code, notes, and snippets.

@AnonymousArthur

AnonymousArthur/Caffe2GPU_Win10_VS2017_CUDA9_cuDNN7.md

Last active April 24, 2018 13:19
Download ZIP

Revisions

  1. AnonymousArthur revised this gist Apr 24, 2018. 1 changed file with 0 additions and 1 deletion.
    1 change: 0 additions & 1 deletion Caffe2GPU_Win10_VS2017_CUDA9_cuDNN7.md
    View file
    Original file line number Diff line number Diff line change
    @@ -12,7 +12,6 @@ Install within a virtual environment is recommended. So it doesn't pollute your
    SET USE_CUDA=ON
    SET CMAKE_GENERATOR="Visual Studio 15 2017 Win64"
    `
    and find DBUILD_PYTHON, override OFF to ON, so it can build python bindings.
    7. Open Visual Studio 2017 Prompt(DO NOT open normal cmd!!!)
    8. Go to the <pytorch source>\scripts\
    9. execute `build_windows.bat` and then you wait (very long time).
  2. AnonymousArthur renamed this gist Apr 23, 2018. 1 changed file with 1 addition and 1 deletion.
    2 changes: 1 addition & 1 deletion README.md → Caffe2GPU_Win10_VS2017_CUDA9_cuDNN7.md
    View file
    Original file line number Diff line number Diff line change
    @@ -15,5 +15,5 @@ SET CMAKE_GENERATOR="Visual Studio 15 2017 Win64"
    and find DBUILD_PYTHON, override OFF to ON, so it can build python bindings.
    7. Open Visual Studio 2017 Prompt(DO NOT open normal cmd!!!)
    8. Go to the <pytorch source>\scripts\
    9. execute `build_windows.bat` and then you wait.
    9. execute `build_windows.bat` and then you wait (very long time).
    10. The finishing up stuff go [Caffe 2](https://caffe2.ai/docs/getting-started.html?platform=windows&configuration=compile) and do it.
  3. AnonymousArthur created this gist Apr 23, 2018.
    19 changes: 19 additions & 0 deletions README.md
    View file
    Original file line number Diff line number Diff line change
    @@ -0,0 +1,19 @@
    Refrence [Caffe 2](https://caffe2.ai/docs/getting-started.html?platform=windows&configuration=compile)
    Assuming python is already installed and you've done this [step](https://caffe2.ai/docs/getting-started.html?platform=windows&configuration=compile#null__regular-python-install)
    Install within a virtual environment is recommended. So it doesn't pollute your system envrionment, and you can always retry.
    1. Download Visual Studio 15.4.5 CE, do not download the latest 15.6.X because CUDA 9.1 doesn't support that.
    2. Launch the VS installer, select development with C++ and also select C++/CLI support and VC++ 2015.3 v140 toolset. DO NOT UNSELECT ANY SELECTED OPTIONS!!!
    3. Download CUDA 9.1 and cuDNN7.1 from Nvidia, install and patch CUDA if necessary, and then put cnDNN stuff into installation folder as instructed. This step must be performed after step 2 complete.
    4. Download and Install CMAKE
    5. Clone the project, it is now a subproject of pyTorch. Use the link: [git clone --recursive https://github.com/pytorch/pytorch.git
    ](git clone --recursive https://github.com/pytorch/pytorch.git)
    6. Go to pytorch\scripts. Modify build_windows.bat, for us:
    `
    SET USE_CUDA=ON
    SET CMAKE_GENERATOR="Visual Studio 15 2017 Win64"
    `
    and find DBUILD_PYTHON, override OFF to ON, so it can build python bindings.
    7. Open Visual Studio 2017 Prompt(DO NOT open normal cmd!!!)
    8. Go to the <pytorch source>\scripts\
    9. execute `build_windows.bat` and then you wait.
    10. The finishing up stuff go [Caffe 2](https://caffe2.ai/docs/getting-started.html?platform=windows&configuration=compile) and do it.