ubuntu16.04 anaconda安装tensorflow-gpu详细教程

1、更新NVIDIA驱动

基于ppa源安装NVIDIA驱动

1 sudo add-apt-repository ppa:graphics-drivers/ppa  
2 sudo apt-get update  

此时打开Software & Updates ，选择 Additioal Drivers，选择要安装的驱动版本，点击右下角的 Apply Changes按钮。

（因为我已安装了对应版本，所以 Apply Changes按钮不可点，未安装时是可点的）

  耐心等待，安装完成后，点击 Revert 按钮（重启电脑）

注意：python3.6对应的Anaconda版本是：Anaconda3-5.2.0-Linux-x86_64.sh 。此版本的Anaconda安装 tensorflow-gpu 时自动安装的CUDA版本是9.2 。依据下表，故NVIDIA的版本需 。

2 cuda  cuddnn 安装

(1) 安装cuda

设置环境变量，在/home/<yourname>/.bashrc文件中追加：

    根据你的安装位置和版本
    export PATH=$PATH:/usr/local/cuda-9.2/bin
    export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-9.2/lib64

(2) 安装cudnn

解压下载的文件，可以看到cuda文件夹，在当前目录打开终端，执行如下命令：

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
sudo chmod a+r /usr/local/cuda/lib64/libcudnn*

(3) 查看cudnn版本

cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2

3、安装Anaconda

去Anaconda官网下载所需的版本，本人下载的时python3.6对应的版本。（python3.6对应的Anaconda版本是：Anaconda3-5.2.0-Linux-x86_64.sh    python3.5对应的Anaconda版本是：Anaconda3-4.2.0-Linux-x86_64.sh ）（tensorflow应该暂不支持python3.7）不同Anaconda版本的安装的tensorflow版本不同（前者越新，安装的tensorflow版本越新），其对应的cuda版本也不同，特别注意版本对应的问题。

 

打开终端，输入如下命令进行安装

bash Anaconda3-5.2.0-Linux-x86_64.sh

中间会有“是否添加环境变量”的选项（如下图），选择 yes

4、使用 Conda 安装tensorflow-gpu

其中将第一句改为：conda create --name tftest python=3.6

第3句改为：conda install tensorflow-gpu

耐心等待安装完成（这个过程比较漫长，慢慢等待）

5、测试tensorflow是否安装成功

  可以通过一个简单的程序测试TensorFlow是否安装成功。直接在终端中键入以下代码：

import tensorflow as tf 
sess = tf.Session() 
a = tf.constant(1) 
b = tf.constant(2) 

print(sess.run(a+b)) 

结果：3

6、PyCharm的配置

打开PyCharm之后，点击右下角的Configure，选择Settings，在左侧选择Project Interpreter，然后点击右侧面板的小齿轮，选择Add...，在弹出的窗口左侧选择Conda Environment，接着选择Existing environment，这时候它会自动识别出刚刚创建好的conda环境，然后勾选Make available to all projects，点击OK即可。

创建新项目的时候注意选择好Project Interpreter，因为刚刚的设置可能会不能应用到新项目，需要手动选择一下。

7、环境中opencv安装：

（1）sudo apt-get install python3-pip

(2）升级pip3)

其实升级pip3只需要两步，1.输入升级指令 2.修改配置文件。

升级指令为：

sudo pip3 install --upgrade pip

注意：如果只进行这一步的话接下来的安装仍会报错，提示错误大致为：cannot import name 'main'。所以还需要修改/usr/bin/pip3文件。

先通过 sudo vim /usr/bin/pip3 打开文件，并将文件修改为

from pip import __main__

if __name__ == '__main__':

    sys.exit(__main__._main())

保存退出后即可完成pip3的更新。/usr/bin/pip3文件如下图所示：

（3）

pip install opencv-python

7、pip  换源

1）检查pip.conf文件是否存在
    >> cd ~
    >> mkdir .pip
    >> ls ~/.pip
2）直接编辑pip.conf
    sudo gedit ~/.pip/pip.conf 

[global]
index-url = http://mirrors.aliyun.com/pypi/simple/
[install]
trusted-host = mirrors.aliyun.com