Ubuntu下安装tensorflow-gpu（1.7）

一、系统环境

操作系统：Ubuntu（16.04）

显卡       ：NVIDIA GeForce GTX 1080ti

二、CUDA安装

1、下载CUDA Toolkit 9.0

      下载地址：https://developer.nvidia.com/cuda-90-download-archive

      也可从以下连接下载：https://developer.nvidia.com/compute/cuda/9.0/Prod/local_installers/cuda_9.0.176_384.81_linux-run

2、安装CUDA

     1）运行：

sudo sh cuda_9.0.176_384.81_linux.run

     2）按照命令行提示操作（我全部选择了y），请确定 X Server已经退出，否则会报错：

     

          X Server 关闭命令：

sudo /etc/init.d/lightdm stop

   3）安装log中提示缺少如下包

      

       原因是缺少相关的依赖库,安装相应库就解决了

sudo apt-get install freeglut3-dev build-essential libx11-dev libxmu-dev libxi-dev libgl1-mesa-glx libglu1-mesa libglu1-mesa-dev 

再次安装,就不再提示了

sudo sh cuda_9.0.176_384.81_linux.run

4）设置环境变量

sudo vi ~/.bashrc

在文本末尾增加如下内容

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

环境变量立即生效 

sudo source ~/.bashrc  
sudo ldconfig 

5）检查cuda是否配置正确

nvcc --version

三、CUDNN安装

1、下载Download cuDNN v7.0.4 (Nov 13, 2017), for CUDA 9.0

      下载地址：https://developer.nvidia.com/cudnn ，需要登录后才能下载，请选择cuDNN v7.0.4版本

2、下载完，解压cudnn,会出现三个文件，如下：

tar  zxvf cudnn-9.0-linux-x64-v7.tgz

   把cudnn两个文件夹中的文件，复制到CUDA相应文件夹的下面（cuda的默认安装路径：/usr/local/cuda）

cp ./include/* /usr/local/cuda/include/
cp ./lib64/* /usr/local/cuda/lib64/

cudnn安装ok

四、安装tensorflow-gpu

1、安装pip  （如果报权限错误，则在命令前加 sudo）

apt-get install -y wget vim
wget -O /tmp/get-pip.py https://bootstrap.pypa.io/get-pip.py
python3 /tmp/get-pip.py
pip install --upgrade pip

2、安装其他libs

RUN  pip install -U numpy
RUN  pip install -U matplotlib
RUN  pip install -U pandas

3、安装tensorflow-gpu  （如果报权限错误，则在命令前加 sudo）

tensorflow的官网：https://tensorflow.google.cn/install/gpu

pip install -U tensorflow-gpu==1.7 --user

 

 

参考

Win10环境+ CUDA9.0 +CUDNN7.0+TensorFlow1.7/1.6/1.5配置

ubuntu tensorflow install（Ubuntu16.04+CUDA9.0+cuDNN7.5+Python3.6+TensorFlow1.5）