解決torch.cuda.is_available為False

問題：在Anaconda環境下，電腦擁有GPU并且已經通過conda install安裝了Pytorch、cudatoolkit，但是torch.cuda.is_available()始終返回False（找不到顯卡）

原因：使用conda install安裝的pytorch、torchvision均為CPU版本，無法使用GPU進行計算。

解決：訪問 鏡像源 下載對應的GPU版本Pytorch、Torchvision
CPU版：

GPU版：

進入Anaconda的對應環境下，使用pip install 下載鏡像路徑名進行安裝

完成后再次調用GPU

import torch device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")# Assuming that we are on a CUDA machine, this should print a CUDA device:print(device)

成功返回gpu:0

總結

以上是生活随笔為你收集整理的解决torch.cuda.is_available()为False的问题的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。