在网上查了很多资料,首先要确定是不是版本问题
,
像我在虚拟环境pip安装用pip list
查看的cuda和torch分别是10.2.89和1.5.0
但是在cuda list
里面是10.2.89和1.0.0
很怪,无法理解
但是在python中打印
出cuda和torch又分别是9.0和1.0.0
5555好窒息,搞了一整天很暴躁
conda list
在python中打印
无法理解
conda list和pip list的版本不同又和python打印出来的不同,而且我配置的torchvision是0.4的,肯定不能支持cuda9.0
具体可参考 https://www.zhihu.com/question/395145313/answer/1260811851
好吧,强行重装cuda和torch
conda install pytorch=1.5.0 cudatoolkit=10.2.89 -c pytorch
再次在python里面打印出来,
import torch
print(torch.__version__)
print(torch.cuda.is_available())
torch.version.cuda
成功!再次运行,报错解除!
最终环境是
python3.7.11
cuda 10.2.89
cudnn 7.6.5
torch 1.5.0
torchvision 0.4