出现的情况:
解决方法:
参考了许多博客
建议的方法:
创建一个虚拟环境
conda create -n torch_1.3 python=3.6
激活虚拟环境
conda activate torch_1.3
conda安装
conda install pytorch==1.5.0
如果报错每个包单独安装就可以了,不要一次性安装多个
附上Pytorch安装命名?Previous PyTorch Versions | PyTorch
如果出现UserWarning: Failed to load image Python extension: [WinError 127] 找不到指定的程序。warn(f"Failed to load image Python extension: {e}")
应该是torch和torchvision版本不匹配的原因
如果出现GPU不可用的情况
import torch
torch.cuda.is_available()
去命令行输入
conda list
看看是不是下载的CPU版本
像我这样就是CPU版本,卸载后下载GPU版本即可
这样就是ok的?