测试的时候提示cudnn的库找不到

[sofa-li]

安装的时候设置了 CUDA_HOME=/usr/local/cuda-12.1
CUDNN_INCLUDE_DIR=/usr/local/cuda-12.1/include
CUDNN_LIB_DIR=/usr/local/cuda-12.1/lib64
python3 setup.py develop

但是跑测试的时候 PYTHONPATH="./:${PYTHONPATH}" CUDA_VISIBLE_DEVICES=0 python3 basicsr/test.py -opt options/test/SRResNet_SRGAN/test_MSRResNet_x4.yml

提示 Could not load library libcudnn_cnn_infer.so.8. Error: libcuda.so: cannot open shared object file: No such file or directory

实际上这个库文件在 lib64里面的

CUDNN_INCLUDE_DIR CUDNN_LIB_DIR 添加到了环境变量也不行