[解決] 安裝pytorch後 torch.cuda.is

Ubuntu安裝pytorch後 torch.cuda.is_available()結果為false

原因：安裝的cuda版本和電腦的驅動不一致而導致的。torch.cuda.is_available()結果為false:在環境窗口輸入命令, 查看去讀版本驅動

<code>
nvcc
 -V
/<code>

下載對應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。