Fix docs for Windows CUDA 8 builds (#9254)

Summary:
Fixes #9200.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/9254

Differential Revision: D8778011

Pulled By: soumith

fbshipit-source-id: 0a2c2863ac1bc515397fc446039db64d1d4e236d
This commit is contained in:
peter 2018-07-09 18:42:56 -07:00 committed by Facebook Github Bot
parent 99ab082366
commit 19ecb5f8ad

View File

@ -216,6 +216,9 @@ set CMAKE_GENERATOR=Visual Studio 15 2017 Win64
set DISTUTILS_USE_SDK=1
REM The following line is needed for Python 2.7, but the support for it is very experimental.
set MSSdk=1
REM As for CUDA 8, VS2015 Update 2 or up is required to build PyTorch. Use the following two lines.
set "PREBUILD_COMMAND=%VS140COMNTOOLS%\..\..\VC\vcvarsall.bat"
set PREBUILD_COMMAND_ARGS=x64
call "%VS150COMNTOOLS%\vcvarsall.bat" x64 -vcvars_ver=14.11
python setup.py install