根据Pytorch我在cmd中使用了这个命令
pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu113
但是CUDA仍然不可用。有人可以帮我吗?
import torch
print(torch.cuda.is_available())
输出将是
False
您仍然需要在计算机上安装 cuda-toolkit 和 cudnn
我也遇到了同样的问题,我的解决方法是卸载 Pytorch 并安装带有 CUDA 12.1 的 Nightlys 版本。我正在使用 CUDA 工具包 12.3。
运行此命令,看看这是否是一个比更新驱动程序更小的问题:
pip uninstall torch
pip3 install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/cu121
只需确保您的 Pytorch 版本支持相应的 CUDA Toolkit 版本,并且两者都不匹配。如果这没有帮助,请参阅 @trsvchn 关于更新驱动程序的回复,或检查您是否满足 系统要求。