运行tenfordlow-gpu时,报错 ImportError: libcudnn.so.6 cannot open shared object file: No such file or director
找不到cudnn的文件,原因可能是1)没有装cudnn;2)cudnn安装成功了,但是没有加到环境变量中去。
我遇到的是第二种情况,于是添加cudnn的路径到环境变量:
sudo gedit ~/.bashrc
export LD_LIBRARY_PATH="$LD_LIBRARY_PATH:/usr/local/cuda-8.0/lib64:/usr/local/cuda-8.0/extras/CUPTI/lib64"
export CUDA_HOME=/usr/local/cuda-8.0
最后再激活环境变量:
source ~/.bashrc