当前位置：网站首页 > 技术文章 > 正文

[解决] 安装pytorch后 torch.cuda.is_available()结果为false

“[解决] 安装pytorch后 torch.cuda.is_available()结果为false” 网站介绍

原因：安装的cuda版本和电脑的驱动不一致而导致的。torch.cuda.is_available()结果为false:在环境窗口输入命令, 查看去读版本驱动

nvcc -V
# 或者 nvidia-smi

下载对应cuda版本的pytorchpytorch官网主页.

如果使用上述指令直接安装，安装的是torch1.5 + cuda9.2/10.1/10.2,很多时候吗，我们电脑的版本可能9.0，甚至更低，直接安装就会用不了cuda。

我们可以点击下面链接：pytorch历史版本.

如我的电脑是9.0版本，那么找到cudatoolkit=9.0的指令下载，此时安装的就是torch1.01+torchvision0.2.2+cudatoolkit9.0

看看结果：

成功了，可是使用cuda。