前提条件:
只需根据自己的系统键入下面的一条命令即可:(注:仅 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
原文:https://www.cnblogs.com/ylHe/p/10871487.html