Skip to content
Discussion options

You must be logged in to vote

解决了,我这里是本机安装了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

Replies: 1 comment

Comment options

You must be logged in to vote
0 replies
Answer selected by SWHL
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Category
Q&A
Labels
None yet
1 participant
Converted from issue

This discussion was converted from issue #10784 on May 25, 2024 05:51.