Torch和cuda和cudnn版本问题

我的cuda版本

C:\Users\Administrator>nvidia-smi
Sun Jul 28 11:55:31 2024
+-----------------------------------------------------------------------------------------+
| NVIDIA-SMI 555.85                 Driver Version: 555.85         CUDA Version: 12.5

cudnn版本 v8.9.7

cuDNN Archive | NVIDIA Developer

我原本安装的torch是torch-2.1.2+cu121-cp310-cp310-win_amd64.whl。

结果无法使用cuda

>>> import torch
>>> torch.cuda.is_available()
>>> False

重新安装torch，版本cu124

(sd-webui) D:\Program\stable-diffusion-webui>pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu124

结果成功了

>> import torch
>>> torch.cuda.is_available()
>>> True