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
1条答案
按热度按时间azpvetkf1#
nvm,我发现,如果我只是不能使用 Torch 11.7,必须使用11.6.对于其他ppl可能会得到同样的问题,卸载每个与 Torch 相关的pkg,而不是使用这个:
使用此选项: