Original Content

Easy Installation

# If your main Python version is not 3.5 or 3.6conda create-n test python=3.6 numpy pyyaml mkl# for CPU only packagesconda install-c peterjc123 pytorch-cpu# for Windows 10 and Windows Server 2016, CUDA 8conda install-c peterjc123 pytorch# for Windows 10 and Windows Server 2016, CUDA 9conda install-c peterjc123 pytorch cuda90# for Windows 7/8/8.1 and Windows Server 2008/2012, CUDA 8conda install-c peterjc123 pytorch_legacy

About CI packages

About the nightly packages

How to find the package and install?

# For all versions# Windows x64# Python x64 3.5 / 3.6# MKL/Numpy/PyYAML# For GPU versions# CUDA 9 / 9.1# cuDNN 7# NVTX (Visual Studio Integration in CUDA. if it fails to be installed, you can extract#       the CUDA installer exe and found the NVTX installer under the CUDAVisualStudioIntegration)pip install numpy mkl intel-openmppip install pytorch-[version]-cp[pyversion]-cp[pyversion]m-win-amd64.whl# Add [PythonRoot]\Library\bin into environment variable `PATH` and restart command prompt before using.

C:\Anaconda2\lib\site-packages\torch\__init__.pyc in <module>()39     os.environ['PATH']= os.path.dirname(__file__)+'\\lib\\;'+ os.environ['PATH']40---> 41     from torch._C import *42__all__+= [namefor nameindir(_C)43if name[0]!='_'andImportError:DLL load failed: The specified procedure could not be found.

Compiling Examples

################################################################################# These are the flags that you can set before running the scripts.################################################################################# You can specify which version you want to build# If you omit it, it will build for the master branch on defaultset PYTORCH_BUILD_VERSION=0.4.1# If you want to build your own fork of PyTorchset PYTORCH_REPO=pytorch# If you want to build for a specific branch# Note: If you specify PYTORCH_BUILD_VERSION too, it won't be used.set PYTORCH_BRANCH=master# If you want to skip the confirmation of the flags configured by our scriptset NO_PROMPT=1# If you want to continue build without a verified VS version (Safe for CPU builds)set SKIP_VS_VER_CHECK=1# If you want to skip prompt for building for Python 2.7 (Not recommended)set SKIP_PY_VER_CHECK=1# If you want to build for LibTorchset BUILD_PYTHONLESS=1################################################################################# These are the available scripts that you can use.################################################################################# If you don't want to override the default settingsauto.bat# If you don't want to compile with CUDAcpu.bat# If you want to compile with CUDA 8cuda80.bat# If you want to compile with CUDA 9cuda90.bat# If you want to compile with CUDA 9.1cuda91.bat# If you want to compile with CUDA 9.2cuda92.bat# If you want to compile with CUDA 10.0cuda100.bat

Using Examples

# The main difference in Python between Windows and Unix systems is multiprocessing# So please refactor your code into the following structure if you use DataLoaderimporttorchdefmain():fori, (x,y)indataloader:# do something hereif__name__=='__main__':main()