项目场景:
在docker中运行tensorflow相关项目RTX3060
cuda 11.2
cudnn 8.0.1
问题描述:
运行时报错Loaded runtime CuDNN library: 8.0.5 but source was compiled with: 8.1.0
原因分析:
开始安装的是cudnn8.1.1,运行报错。改用cudnn8.1.0后,仍然报错。 cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2
查看cuda,cudnn以及tensoflow适配的cuda版本都无问题。
解决方案:
在docker中修改cuda路径vim ~/.bashrc
添加cuda相关路径,解决报错
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/lib64
export PATH=/usr/local/cuda/bin:$PATH
export CUDA_HOME=/usr/local/cuda
在这个问题上找了很久,都说是cuda版本的问题。但是安装后依然无法调用到。猜测调用的不是安装的cuda,遂修改。具体原因还没去了解。之前安装cuda是总会设置路径,docker中没有cuda路径,不知道是如何调用cuda的。