2021-03-17

关于使用tensorflow时,cuda和cudnn不匹配问题的解决方案


利用deepC训练模型时,爆出来很多错误,将缺失的文件一一备齐后还是出现如下的报错:



刚开始认为是tensorflow的版本问题,重新用conda建了一个新的环境,将tensorflow的版本降低为2.7

pip install tensorflow==2.7.0

还是没能解决问题,然后根据报错提示,怀疑是cuda和CuDNN的匹配问题,因此从CuDNN下面官网下载了与cuda-11.0相匹配的cudnn-v8.1.0.77版本

https://developer.nvidia.com/rdp/cudnn-archive


解压文件,并进入相应目录,运行如下命令:

sudo cp cuda/includes/cudnn.h /usr/local/cuda-11.1/include

sudo cp cuda/lib64/libcudnn* /usr/local/cuda-11.1/lib64

sudo chmod a+r/usr/local/cuda-11.1/include/cudnn.h

sudo chmod a+r/usr/local/cuda-11.1/lib64/libcudnn*


重新运行程序,居然跑通了,妈妈咪呀,两天时间啊,终于解决了

你可能感兴趣的:(2021-03-17)