TensorFlow最新版本目前好像不支持cuda8.0，也不支持cuda10，好不容易终于升级了cuda9.0，终于安装了TensorFlow1.11.0。特此记录一下，服务器Ubuntu14.04.5。
- 在CUDA Toolkit下载对应版本Cuda9.0的.run文件。
- 根据提示安装cuda
sudo chmod +x cuda_9.0.176_384.81_linux.run # 为 cuda_9.0.176_384.81_linux.run 添加可执行权限
./cuda_9.0.176_384.81_linux.run # 安装 cuda_9.0.176_384.81_linux.run
Do you accept the previously read EULA?
accept/decline/quit: accept
Install NVIDIA Accelerated Graphics Driver for Linux-x86_64 384.81?
(y)es/(n)o/(q)uit: n # 如果已经安装过了，则不需要安装显卡驱动
Install the CUDA 9.0 Toolkit?
(y)es/(n)o/(q)uit: y
Enter Toolkit Location
[ default is /usr/local/cuda-9.0 ]: # 选择默认即可
/usr/local/cuda-9.0 is not writable.
Do you wish to run the installation with 'sudo'?
(y)es/(n)o: y
Please enter your password:
Do you want to install a symbolic link at /usr/local/cuda? # 添加软链接便于切换cuda
(y)es/(n)o/(q)uit: y
Install the CUDA 9.0 Samples?
(y)es/(n)o/(q)uit: n
- 在NVIDIA cuDNN注册登录并下载cudnn。
- 安装cndnn
sudo tar -xzvf cudnn-8.0-linux-x64-v5.1.tgz
sudo cp cuda/include/cudnn.h /usr/local/cuda/include
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64
sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*
- sudo vim ~/.bashrc 在文件中添加如下内容，cuda表示之前添加的软链接。
export PATH=/usr/local/cuda/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
export CUDA_HOME=/usr/local/cuda
- source ~/.bashrc 使之生效
- pip install tensorflow-gpu
如果运行命令nvidia-smi出现：
Failed to initialize NVML: Driver/library version mismatch
可以重启服务器，如果不行，重新安装驱动。
如果TensorFlow运行时出现以下两种错误：
CUDA driver version is insufficient for CUDA runtime version
libnvidia-fatbinaryloader.so.381.22: cannot open shared object file: No such file or directory
可以重新安装驱动：
sudo add-apt-repository ppa:graphics-drivers/ppa
sudo apt-get update
sudo apt-get install nvidia-375
如需切换cuda，执行：
rm -rf /usr/local/cuda # 删除之前创建的软链接
sudo ln -s /usr/local/cuda-8.0/ /usr/local/cuda/
nvcc --version # 查看当前 cuda 版本
另外，pycharm远程连接服务器时，可能找不到环境变量，在Run/Edit Configurations/下
添加Environment variables：
LD_LIBRARY_PATH=/usr/local/cuda/lib64
参考自以下博客：
https://blog.csdn.net/Maple2014/article/details/78574275
https://blog.csdn.net/qq_34374211/article/details/81018320
https://www.cnblogs.com/go-better/p/7625541.html
https://stackoverflow.com/questions/43022843/nvidia-nvml-driver-library-version-mismatch
https://blog.csdn.net/seymour163/article/details/78798419
https://www.jianshu.com/p/385a408024d1