出现问题:
cuda版本与驱动版本相对应。
cuda已经安装成功,但是程序调用时失败。
问题原因:
pytorch版本与cuda版本不匹配
比如:
本机环境是cuda 10.1
虚拟环境中一开始安装的pytorch 1.3.1但是对应的是cuda9.2
这个时候就要强制更新pytorch版本为适配版本。
解决方法:
首先进入pytorch官网
首页的版本安装命令为:
当前稳定版本
图片最下方有之前的历史版本,点击可进入查看并挑选对应版本pytorch安装。
找到你(环境需要的&&cuda版本对应)的pytorch安装命令。
如果实在找不到还有一个解决方法:
先去下面的网址看你有没有你需要的
pytorch官方wheel下载存储
用下面命令安装
pip3 install torch===1.3.1 torchvision===0.4.2 -f https://download.pytorch.org/whl/torch_stable.html
torch和torch vision都改成你需要的,配套的。
备注:这个命令是官网没写的pytorch1.3.1的安装命令,我的cuda是10.1,可以适配
最后:
安装成功之后再进入python看一下调用
python
import torch
print(torch.__version__)
print(torch.cuda.is_available())