python-3.x 为什么torch.cuda.is_available()返回False?

wfveoks0  于 2022-11-19  发布在  Python
关注(0)|答案(1)|浏览(163)

torch.cuda.is_available()返回False。安装PyTorch时使用:

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

已尝试安装CUDA 11.7和11.8。环境详细信息:

  • RTX 2080系列
  • 英伟达-SMI 522.06
  • 驱动程序版本:522.06
  • CUDA版本:11.8

pip list的版本:

torch       1.13.0
torchaudio  0.13.0
torchinfo   1.7.1
torchvision 0.14.0
azpvetkf

azpvetkf1#

nvm,我发现,如果我只是不能使用 Torch 11.7,必须使用11.6.对于其他ppl可能会得到同样的问题,卸载每个与 Torch 相关的pkg,而不是使用这个:

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

使用此选项:

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

相关问题