我的环境:
CPU: i5-6300HQ OS:window10 GPU:GTX960
pytorch-cpu :1.4
torchvision: 0.5
首先win+R打开cmd窗口,输入命令
nvcc --version
我的cuda版本是10.0。好,现在我们打开pytorch官网,选择对应cuda版本的pytorch生成下载命令。
重点来了!
我们安装过pytorch的朋友应该知道,用pip从官网下载是很慢的,网上很多教程是建议换清华镜像下载,但还是很慢!这里教大家一个技巧:
我们复制刚刚产生的pip下载命令到cmd窗口运行
接下来按下CRTRL+C停止下载,我们可以看到这个whl文件真正的下载链接:
https://download.pytorch.org/whl/cu101/torch-1.4.0-cp37-cp37m-