解決Linux下無法使用GPU，torch.cuda.is_available()返廻False的問題（無論怎麽下載都是cpu版本的問題）_torch無法使用gpu_H1tpp的博客-CSDN博客

首先說明，很多博主文章裡沒有提到的一點，pytorch包和torchvision包是一同下載的，且pytroch包包含在torchvision內，因此要先把環境中的pytorch，torchvision，torchaudio都一一卸載掉

使用命令

pip uninstall torchvision pytorch torchaudio torch

之後用nvidia-smi查看對應cuda 版本

然後使用這個命令

pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu113

後麪的113換成你的cuda版本對應的（我的是11.3所以就是113）

這樣下載下來的就不是cpu版本的了 

我們來check一下，命令行鍵入

python

import torch

print(torch.cuda.is_available())