-
Beta Was this translation helpful? Give feedback.
Answered by
keyfall
Sep 1, 2023
Replies: 1 comment
-
解决了,我这里是本机安装了cuda和cudnn,conda安装了cuda和cudnn,我这里tools/train.py可以,tools/infer/predict_det.py就会报上述错误,这里我发现conda虚拟环境运行命令时使用的cuda一直是本地的cuda,然后cudnn用的是虚拟环境的cudnn,直接说结果 还有个可能是cudnn的版本不对,虽然上面显示支持11.*,但是还是会显示cudnn_status_not_supported,降低版本就好,我使用了8.9不行,一个一个降低,到8.4.1.50时可以使用了 |
Beta Was this translation helpful? Give feedback.
0 replies
Answer selected by
SWHL
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
解决了,我这里是本机安装了cuda和cudnn,conda安装了cuda和cudnn,我这里tools/train.py可以,tools/infer/predict_det.py就会报上述错误,这里我发现conda虚拟环境运行命令时使用的cuda一直是本地的cuda,然后cudnn用的是虚拟环境的cudnn,直接说结果
修改环境变量,我这里加上了一个/usr/local/cuda-11.7/lib64,使用了这个应该就会把设置的cuda设置为一直调用的cuda,所以把这句删了,就好了
还有个可能是cudnn的版本不对,虽然上面显示支持11.*,但是还是会显示cudnn_status_not_supported,降低版本就好,我使用了8.9不行,一个一个降低,到8.4.1.50时可以使用了
conda install cudnn=8.4.1.50 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge