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按钮。

ubuntu16.04 anaconda安装tensorflow-gpu详细教程_第1张图片

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

  耐心等待,安装完成后,点击 Revert 按钮(重启电脑)

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

ubuntu16.04 anaconda安装tensorflow-gpu详细教程_第2张图片

ubuntu16.04 anaconda安装tensorflow-gpu详细教程_第3张图片

2 cuda  cuddnn 安装

(1) 安装cuda

设置环境变量,在/home//.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版本也不同,特别注意版本对应的问题。

 

ubuntu16.04 anaconda安装tensorflow-gpu详细教程_第4张图片

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

bash Anaconda3-5.2.0-Linux-x86_64.sh

中间会有“是否添加环境变量”的选项(如下图),选择 yes

ubuntu16.04 anaconda安装tensorflow-gpu详细教程_第5张图片

4、使用 Conda 安装tensorflow-gpu

ubuntu16.04 anaconda安装tensorflow-gpu详细教程_第6张图片

其中将第一句改为: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文件如下图所示:

ubuntu16.04 anaconda安装tensorflow-gpu详细教程_第7张图片

(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

 

你可能感兴趣的:(机器学习,深度学习)