[已解决]torch.cuda.is_available()结果为false

@TOC

Ubuntu安装pytorch后 torch.cuda.is_available()结果为false

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

nvcc -V

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

如果使用上述指令直接安装，安装的是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。