蒸腾了两天,终于搞定了
是cudnn版本的问题
更新cudnn的时候,首先要删除/usr/local/cuda-10.0/targets/x86_64-linux/lib路径下所有之前cudnn版本的库,sudo rm libcudnn*
然后 将下载的新版本的cudnn拷贝到cuda的目录,命令如下
# 复制cudnn头文件
sudo cp cuda/include/* /usr/local/cuda-10.0/include/
# 复制cudnn的库
sudo cp cuda/lib64/* /usr/local/cuda-10.0/lib64/
# 添加可执行权限
sudo chmod +x /usr/local/cuda-10.0/include/cudnn.h
sudo chmod +x /usr/local/cuda-10.0/lib64/libcudnn*
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
3.执行sudo ldconfig -v | grep cudnn 更新库 ,并查看,如果有多条记录就删除多余的