@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。