轉(zhuǎn)載https://blog.csdn.net/qq_36368388/article/details/97927154
報錯如下:
解決辦法:
- 確定cuda以及cudnn的版本
參考:https://blog.csdn.net/wanzhen4330/article/details/81699769
根據(jù)cuda的版本下載對應(yīng)的cudnn安裝包 - 查看cudnn是否成功安裝
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
- 再次執(zhí)行程序庸论,查看是否報錯登钥,如果仍然報錯介时,執(zhí)行命令:
cp /usr/local/cuda-10.0/lib64/libcublas.so.10.0 /usr/local/lib/libcublas.so.10.0 && sudo ldconfig
cp /usr/local/cuda-10.0/lib64/libcufft.so.10.0 /usr/local/lib/libcufft.so.10.0 && sudo ldconfig
根據(jù)報錯的庫,缺少什么就補(bǔ)什么