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

wfveoks0 于 2022-11-19 发布在 Python

关注(0)|答案(1)|浏览(164)

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。环境详细信息：

pip list的版本：

torch       1.13.0
torchaudio  0.13.0
torchinfo   1.7.1
torchvision 0.14.0

1条答案

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