mirror of
https://github.com/saymrwulf/pytorch.git
synced 2026-05-14 20:57:59 +00:00
7b8c7b11d2
Summary: After this, we should have contbuild guarding the Windows build both with and without CUDA. This includes a series of changes that are needed to make Windows build, specifically: (1) Various flags that are needed in the cmake system, specially dealing with /MD, /MT, cuda, cudnn, whole static linking, etc. (2) Contbuild scripts based on appveyo. (3) For Windows build, note that one will need to use "cmake --build" to build stuff so that the build type is consistent between configuration and actual build. see scripts\build_windows.bat for details. (4) In logging.h, ERROR is already defined by Windows. I don't have a good solution now, and as a result, LOG(ERROR) on windows is going to be LOG(INFO). (5) variable length array is not supported by MSVC (and it is not part of C++ standard). As a result I replaced them with vectors. (6) sched.h is not available on Windows, so akyrola 's awesome simple async net might encounter some slowdown due to no affinity setting on Windows. (7) MSVC has a Closes https://github.com/caffe2/caffe2/pull/183 Reviewed By: ajtulloch Differential Revision: D4657831 Pulled By: Yangqing fbshipit-source-id: 070ded372ed78a7e3e3919fdffa1d337640f146e
30 lines
No EOL
1.3 KiB
Batchfile
30 lines
No EOL
1.3 KiB
Batchfile
:: Installation scripts for appveyor.
|
|
|
|
@echo Downloading CUDA toolkit 8 ...
|
|
|
|
appveyor DownloadFile ^
|
|
https://developer.nvidia.com/compute/cuda/8.0/prod/local_installers/cuda_8.0.44_windows-exe ^
|
|
-FileName cuda_8.0.44_windows.exe
|
|
appveyor Downloadfile ^
|
|
http://developer.download.nvidia.com/compute/redist/cudnn/v5.1/cudnn-8.0-windows10-x64-v5.1.zip ^
|
|
-FileName cudnn-8.0-windows10-x64-v5.1.zip
|
|
|
|
@echo Installing CUDA toolkit 8 ...
|
|
cuda_8.0.44_windows.exe -s compiler_8.0 cublas_8.0 cublas_dev_8.0 cudart_8.0 curand_8.0 curand_dev_8.0 nvrtc_8.0 nvrtc_dev_8.0
|
|
set PATH=%ProgramFiles%\NVIDIA GPU Computing Toolkit\CUDA\v8.0\bin;%ProgramFiles%\NVIDIA GPU Computing Toolkit\CUDA\v8.0\libnvvp;%PATH%
|
|
:: TODO: we will still need to figure out how to install cudnn.
|
|
7z x cudnn-8.0-windows10-x64-v5.1.zip
|
|
copy cuda\include\cudnn.h ^
|
|
"C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\include\"
|
|
copy cuda\lib\x64\cudnn.lib ^
|
|
"C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\lib\x64\"
|
|
copy cuda\bin\cudnn64_5.dll ^
|
|
"C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\bin\"
|
|
|
|
:: Make sure that nvcc is working correctly.
|
|
nvcc -V || exit /b
|
|
|
|
:: Miniconda path for appveyor
|
|
set PATH=C:\Miniconda-x64;C:\Miniconda-x64\Scripts;%PATH%
|
|
:: Install numpy
|
|
conda install -y numpy |