运行 TensorFlow 程序时导入错误:libcudnn

2024-02-28

我在尝试运行 TensorFlow 程序时遇到以下错误:



ImportError: libcudnn.Version: cannot open shared object file: No such file or director
  

Just 下载cuDNN 5.1 https://developer.nvidia.com/compute/machine-learning/cudnn/secure/v5.1/prod_20161129/8.0/cudnn-8.0-linux-x64-v5.1-tgz并按照步骤进行操作(在 Ubuntu 16.04、CUDA 工具包 8.0 上测试)

$ tar xvzf cudnn-8.0-linux-x64-v5.1-ga.tgz
$ sudo cp -P cuda/include/cudnn.h /usr/local/cuda/include
$ sudo cp -P cuda/lib64/libcudnn* /usr/local/cuda/lib64
$ sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*

现在设置路径变量

$ vim ~/.bashrc

export LD_LIBRARY_PATH="$LD_LIBRARY_PATH:/usr/local/cuda/lib64:/usr/local/cuda/extras/CUPTI/lib64"
export CUDA_HOME=/usr/local/cuda

and done

欲了解更多详细信息,您可以查看此site https://medium.com/@ikekramer/installing-cuda-8-0-and-cudnn-5-1-on-ubuntu-16-04-6b9f284f6e77

本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)

运行 TensorFlow 程序时导入错误:libcudnn 的相关文章

随机推荐