如果有GPU,需要按照CUDA和CuDNN
没有,直接从https://download.pytorch.org/whl/torch_stable.html,根据选择cpu版,win操作系统,下载whl文件。
安装whl报错
检查一下python版本是不是对不上
我的python是3.7版本,所以下载链接要选带cp37的,而不是cp36.
torch版本也选择一个新一点的torch-1.5.0+cpu-cp37-cp37m-win_amd64.whl
安装成功!
pip list检查一下,已经有了
torchvision用同样的方法安装一下。
我使用spyder作为IDE,到这一步已经可以使用了,import torch代码执行成功。
如果使用的不是spyder, 也可以选择在pycharm中,创建虚拟环境,激活虚拟环境后,进行whl的安装。
再给当前py项目关联虚拟环境。在文件--设置--项目解释器中,添加上面创建的解释器,就可以关联上了。
import pytorch就可以用了