build_pytorch.bat

set PATH=C:\Program Files\CMake\bin;C:\Program Files\7-Zip;C:\curl-7.57.0-win64-mingw\bin;C:\Program Files\Git\cmd;C:\Program Files\Amazon\AWSCLI;%PATH%
:: Install MKL
curl -k https://s3.amazonaws.com/ossci-windows/mkl.7z -o mkl.7z && 7z x -aoa mkl.7z -omkl
set LIB=%cd%\mkl;%LIB%
:: Install MAGMA
curl -k https://s3.amazonaws.com/ossci-windows/magma_cuda90_release.7z -o magma_cuda90_release.7z && 7z x -aoa magma_cuda90_release.7z -omagma_cuda90_release
set MAGMA_HOME=%cd%\magma_cuda90_release
:: Install clcache
curl -k https://s3.amazonaws.com/ossci-windows/clcache.7z -o clcache.7z && 7z x -aoa clcache.7z -oclcache

call "C:\Program Files (x86)\Microsoft Visual Studio\2017\Community\VC\Auxiliary\Build\vcvarsall.bat" x86_amd64

set PATH=C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\bin;C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\libnvvp;%PATH%
set CUDA_PATH=C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0
set CUDA_PATH_V9_0=C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0
set NVTOOLSEXT_PATH=C:\Program Files\NVIDIA Corporation\NvToolsExt
set CUDNN_LIB_DIR=C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\lib\x64
set CUDA_TOOLKIT_ROOT_DIR=C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0
set CUDNN_ROOT_DIR=C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0
set TORCH_CUDA_ARCH_LIST=5.2
set USE_CLCACHE=1
set CLCACHE_DIR=%cd%\clcache_tmp
set CC=%cd%\clcache\clcache_main.exe
set CXX=%cd%\clcache\clcache_main.exe
set DISTUTILS_USE_SDK=1
set CMAKE_GENERATOR=Ninja
xcopy /Y aten\src\ATen\common_with_cwrap.py tools\shared\cwrap_common.py

python setup.py install