torch版本问题导致 cuad加速失败

[Killua67]

出错信息
Could not locate cudnn_ops64_9.dll. Please make sure it is in your library path!
Invalid handle. Cannot load symbol cudnnCreateTensorDescriptor

复现步骤

1. 使用cpu无问题，勾选CUDA加速报错

操作系统
win11
版本：
nvidia-smi
Fri Oct 25 21:17:42 2024
+-----------------------------------------------------------------------------------------+
| NVIDIA-SMI 566.03 Driver Version: 566.03 CUDA Version: 12.7 |

nvcc -V
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2023 NVIDIA Corporation
Built on Wed_Nov_22_10:30:42_Pacific_Standard_Time_2023
Cuda compilation tools, release 12.3, V12.3.107
Build cuda_12.3.r12.3/compiler.33567101_0

软件左上角版本号数字
windows 2.9 源码安装

解决方案
因安装教程中限制2.2.0 导致找不到文件，取消限制，安装最新版 2.5.0+cu124 正常使用

pip uninstall -y torch torchaudio
pip install torch torchaudio --index-url https://download.pytorch.org/whl