ubunutu配置tensorflow-gpu

我电脑的配置：ubuntu18.04 + i7-9700k + rtx-titan

要安装的：tensorflow-gpu1.14.0

(1) 配置CUDA+cuDnn

tensorflow1.14 适配cuda10.0 + cudnn7.4

https://developer.nvidia.com/cuda-toolkit-archive

下载对应的CUDA版本，选择runfile(local),然后进入下载的目录

sudo sh xxx.run进入安装

安装协议可以使用q跳过，显示第一个问题询问是否安装驱动。因为前面我们已经安装了驱动，所以我们不安装，其他问题直接点y即可。

如果安装完出现

Missing recommended library: libXi.so

Missing recommended library: libXmu.so

等几个lib没安装，使用如下命令：

sudo apt-get install nvidia-modprobe freeglut3-dev libx11-dev libxmu-dev libxi-dev libglu1-mesa-dev

直接apt安装补全包，然后重启，重启之后打开终端卸载CUDA

cd /usr/local/cuda/bin/（安装路径）sudo ./uninstall_cuda_10.0.pl然后重新安装CUDA

接下来修改配置环境变量

vim ~/.bashrc ,修改用户的环境变量，在文章最后添加

export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-10.0/lib64
export PATH=$PATH:/usr/local/cuda-10.0/bin
export CUDA_HOME=$CUDA_HOME:/usr/local/cuda-910.0

打开终端输入nvcc-V，会出现CUDA信息即可，表示安装成功

安装cnDNN

下载cudnn7.4,注意下载deb文件类型的，进入路径安装：

sudo dpkg -i libcudnnxxxxx.deb

即可自动配置好cudnn.

（2）安装tensorflow-gpu(使用豆瓣源)

pip install tensorflow-gpu==1.14.0 -i https://pypi.doubanio.com/simple

等待安装成功测试即可。