Windows上Pytorch安装有3个方案:
- Pytorch官网安装
(-c pytorch)直接https://pytorch.org/(Pytorch官网,按需求选择cuda,py版本,按正常人思维选择会自动给出代码,复制执行即可),其中本地cuda版本查看方式:桌面右键NVIDIA控制面板->左下角系统信息->组件->NVCUDA.DLL右边就是了 - 上面基本不行,国内通过清华源下载,配置源后去除-c pytorch,我试了也不行
- 本地下载whl文件:试了可行,通过第1步基本就知道自己要下上面版本的torch和torchvision,两个要搭配用的,都要知道自己需要的版本
第3个方法的具体步骤
- 在点击有torch和torchvision版本的选择下载 ,选择下载需要的
- 打开cmd或者conda环境窗口,
pip install D:\文件名.whl
自动下载 - 在conda list或者pip list里能看到下载成功的库
我的torch和torchvision通过第1步观察到需要的两个包名是:
torch-1.2.0+cu92-cp37-cp37m-win_amd64.whl
torchvision-0.4.0+cu92-cp37-cp37m-win_amd64.whl
因为要和cuda对应以及两个包要搭配使用,费了很多劲,记录下方便查看。
pass1:如果下载库时候很慢甚至停止,再次打开会飞快~
pass2:记得我下载完运行时候显示PILLOW库版本不对,用pip install PILLOW=xxx
降了版本的
最关键的链接给出来:https://download.pytorch.org/whl/torch_stable.html