蒸騰了兩天,終於搞定了
是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 更新庫 ,並查看,如果有多條記錄就刪除多余的