前提条件:
- Anaconda3 x64 (with Python 3.5/3.6)
- Windows 64位系统(Windows 7 或 Windows Server 2008 及以上)
- GPU版本还需要任意版本的 CUDA (包内置了CUDA 8 / 9 的部分主要二进制文件)
只需根据自己的系统键入下面的一条命令即可:(注:仅 0.3.1,以后不再更新)
# for CPU only packages
conda install -c peterjc123 pytorch-cpu
# for CUDA 8, Windows 10 or Windows Server 2016
conda install -c peterjc123 pytorch
# for CUDA 9, Windows 10 or Windows Server 2016
conda install -c peterjc123 pytorch cuda90
# for CUDA 9.1, Windows 10 or Windows Server 2016
conda install -c peterjc123 pytorch cuda91
# for CUDA 8, Windows 7 or Windows Server 2008/2012
conda install -c peterjc123 pytorch cuda80
# for CUDA 9, Windows 7 or Windows Server 2008/2012
conda install -c peterjc123 pytorch cuda90