Ubuntu16.04 使用anaconda安装TensorFlow1.9(gpu)+cuda9.0+cudnn7.0

ubuntu16.04安装tensorflow1.9(gpu)版本所需软件:nvidia-415,cuda9.0,cudnn7.0,anaconda3-4.4.0
软件下载:
(1)cuda9.0下载地址:
https://developer.nvidia.com/cuda-90-download-archive?target_os=Linux&target_arch=x86_64&target_distro=Ubuntu&target_version=1604&target_type=deblocal
Ubuntu16.04 使用anaconda安装TensorFlow1.9(gpu)+cuda9.0+cudnn7.0_第1张图片
ps:后续cuda的安装步骤按照官网下载地址页面的代码进行操作
(2)cudnn7.0下载地址:https://developer.nvidia.com/cudnn
进入网址之后需要注册账号,然后才能下载
Ubuntu16.04 使用anaconda安装TensorFlow1.9(gpu)+cuda9.0+cudnn7.0_第2张图片
(3)anaconda3-4.4.0下载地址:https://www.anaconda.com/download/
选择python3.6版本。若官网版本已更新,则在网页搜索

一.安装nvidia显卡驱动
1.卸载系统里的Nvidia低版本显卡驱动

sudo apt-get purge nvidia*

2.把显卡驱动加入PPA

sudo -E add-apt-repository ppa:graphics-drivers
sudo apt-get update

3.查找显卡驱动的版本并安装
使用终端命令查看Ubuntu推荐的驱动版本:

ubuntu-drivers devices

(本机是nvidia-415)
采用apt-get命令在终端安装:

sudo apt-get install nvidia-415 nvidia-settings nvidia-prime

4.重启系统并验证
在终端输入以下命令行:
方法1:

lsmod | grep nvidia

如果没有输出,则安装失败。
方法2:查看nvidia版本

nvidia-smi

输入命令行,查看Ubuntu自带的nouveau驱动是否运行:

lsmod | grep nouveau

如果终端没有内容输出,则显卡驱动的安装成功

二 安装cuda 9.0
1.在安装包路径打开终端,输入以下命令(在前面cuda9.0下载地址页面有显示操作代码):

sudo dpkg-i-cuda-repo-ubuntu-9-0-local_9.0.176-1_amd64.deb
sudo apt-key add /var/cuda-repo-9-0-local/7fa2af80.pub
sudo apt-get update
sudo apt-get install cuda

2.配置环境变量
运行如下命令打开profile文件

sudo gedit /etc/profile

打开文件后在文件末尾添加路径,也就是安装目录(根据自己的路径),命令如下:

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

保存,然后重启电脑

sudo reboot

3.验证cuda是否安装成功
方法1:
输入以下命令:

nvcc –V

或者 nvcc --version
方法2:查看cuda版本

cat /usr/local/cuda/version.txt

三、安装cudnn
安装cudnn比较简单,就是复制几个文件:库文件和头文件。将cudnn的头文件复制到cuda安装路径的include路径下,将cudnn的库文件复制到cuda安装路径的lib64路径下。
1.在安装包路径打开终端,执行下面的指令安装:

   tar -zxvf cudnn-9.0-linux-x64-v7.0.tgz
   cd cuda
   sudo cp lib64/lib* /usr/local/cuda/lib64/
   sudo cp include/cudnn.h /usr/local/cuda/include/

然后更新网络连接:

    cd /usr/local/cuda/lib64/  
    sudo chmod +r libcudnn.so.7.0.5  //自己查看.so的版本对应更改
    sudo ln -sf libcudnn.so.7.0.5 libcudnn.so.7
    sudo ln -sf libcudnn.so.7 libcudnn.so  
    sudo ldconfig

2.检验cudnn是否安装成功
方法1:

nvcc –V

方法2:查看cudnn版本

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

四 安装anaconda
1 使用命令:

bash Anaconda3-4.4.0-Linux-x86_64.sh

2 验证是否安装成功
方法1:

anaconda –V

方法2:查看conda版本

conda –V

五 安装TensorFlow
1 输入以下命令安装TensorFlow1.9

conda create -n tensorflow
source activate tensorflow
pip install --ignore-installed --upgrade https://download.tensorflow.google.cn/linux/gpu/tensorflow_gpu-1.9.0-cp36-cp36m-linux_x86_64.whl

安装完成后,输入以下指令关闭TensorFlow:

source deactivate

2 验证是否安装成功
方法1:
终端输入python,进入python命令行后输入:

import tensorflow as tf

没有报错表示安装成功
方法2:查看TensorFlow版本
输入以下指令:

python
import tensorflow as tf
tf.__version__

你可能感兴趣的:(tensorflow,ubuntu)