本文参考了:https://zhuanlan.zhihu.com/p/26871672
非常感谢知乎大佬们提供的工具。
首先要满足:
在开始菜单下找到anaconda所在的文件夹,打开后那个黑色图标就是。
step2. 在Anaconda Prompt下根据自己电脑的版本输入如下命令:
# for CPU only packages
conda install -c peterjc123 pytorch
# for Windows 10 and Windows Server 2016, CUDA 8
conda install -c peterjc123 pytorch cuda80
# for Windows 10 and Windows Server 2016, CUDA 9
conda install -c peterjc123 pytorch cuda90
# for Windows 7/8/8.1 and Windows Server 2008/2012, CUDA 8
conda install -c peterjc123 pytorch_legacy cuda80
剩下的时间,就干点儿别的,一会儿就发现大功告成啦。