frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add windows doc (#5859)

Browse Source
This commit is contained in:
peterjc123
2018-03-23 23:54:44 +08:00
committed by Edward Z. Yang
parent 8e22ef0cb2
commit 03495137d0
1 changed files with 18 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
README.md
View File

@ -164,6 +164,8 @@ If you want to compile with CUDA support, install
If you want to disable CUDA support, export environment variable `NO_CUDA=1`.
If you want to build on Windows, Visual Studio 2017 and NVTX are also needed.
#### Install optional dependencies
On Linux
@ -182,6 +184,11 @@ On macOS
export CMAKE_PREFIX_PATH=[anaconda root directory]
conda install numpy pyyaml setuptools cmake cffi typing
```
On Windows
```cmd
conda install numpy pyyaml setuptools cmake cffi typing
```
#### Get the PyTorch source
```bash
git clone --recursive https://github.com/pytorch/pytorch
@ -199,6 +206,17 @@ On macOS
MACOSX_DEPLOYMENT_TARGET=10.9 CC=clang CXX=clang++ python setup.py install
```
On Windows
```cmd
xcopy /Y aten\src\ATen\common_with_cwrap.py tools\shared\cwrap_common.py
set "VS150COMNTOOLS=C:\Program Files (x86)\Microsoft Visual Studio\2017\Enterprise\VC\Auxiliary\Build"
set CMAKE_GENERATOR=Visual Studio 15 2017 Win64
call "%VS150COMNTOOLS%\vcvarsx86_amd64.bat"
python setup.py install
```
### Docker image
Dockerfile is supplied to build images with cuda support and cudnn v7. Build as usual