龙空技术网

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

喜欢看CBA的码农 135

前言:

如今小伙伴们对“ubuntu查看cuda版本”都比较注意,你们都想要剖析一些“ubuntu查看cuda版本”的相关资讯。那么小编同时在网上汇集了一些对于“ubuntu查看cuda版本””的相关知识,希望同学们能喜欢,朋友们一起来了解一下吧!

Ubuntu安装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。

标签: #ubuntu查看cuda版本