版权声明：本文为博主原创文章，未经博主允许不得转载。 https://blog.csdn.net/sparkexpert/article/details/82953077
最近需要用GPU来进行tensorflow进行训练，发现需要安装cuda 9.0版本才可以，因为tensorflow默认的就是cuda 9.0，为了减少对tensorflow源码进行重构的时间，将之前的cuda8.0版本进行了卸载，然后再安装cuda 9.0
安装过程如下：
# instructions from https://developer.nvidia.com/cuda-downloads (linux -> x86_64 -> Ubuntu -> 16.04 -> deb)
CUDA_REPO_PKG="cuda-repo-ubuntu1604-9-0-local_9.0.176-1_amd64-deb"
wget https://developer.nvidia.com/compute/cuda/9.0/Prod/local_installers/${CUDA_REPO_PKG}
sudo dpkg -i ${CUDA_REPO_PKG}
sudo apt-key adv --fetch-keys http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1604/x86_64/7fa2af80.pub
sudo apt-get update
sudo apt-get -y install cuda-9-0不过安装完成之后，发现tensorflow-gpu还是不管用！报了这个错误：
ImportError: libcudnn.so.7: cannot open shared object file: No such file or directory
定位到/usr/local/cuda目录下，才发现没有这个文件。
在查找相关资料后，可以安装单独的libcudnn 的包，其处理如下：
CUDA_PATCH1="cuda-repo-ubuntu1604-9-0-local-cublas-performance-update_1.0-1_amd64-deb"
wget https://developer.nvidia.com/compute/cuda/9.0/Prod/patches/1/${CUDA_PATCH1}
sudo dpkg -i ${CUDA_PATCH1}
sudo apt-get update安装完成后，发现/usr/local/cuda目录下，仍然没有这个文件。
才发现安装后目录到了/usr/lib/x86_64-linux-gnu这个路径下。
于是进行拷贝：sudo cp libcud* /usr/local/cuda/lib64
再次进入tensorflow，发现报错如下：
这个问题在官网上看到的是tensorflow-gpu 1.8.0的问题，于是升级到1.9.0，再次实验，果然顺利成功。如上图。
扫描二维码关注公众号，回复：
3463460 查看本文章
