Ubuntu 16.04 使用pip命令安装 tensorflow

参考：

https://tensorflow.google.cn/install/install_linux

目的

基于python 3.5，采用pip的方式在Ubuntu 16.04 上安装tensorflow （当前版本为1.11）。

环境要求：

Ubuntu 16.04 或更高版本

CUDA 9.0

cuDNN v7.0   (注意：不要安装v7.1)

主要步骤

一、安装显卡驱动

二、安装CUDA 9.0

三、安装cuDNN v7

下载并解压cuDNN v7，在终端输入以下命令执行安装（要注意当前终端的路径）：

sudo cp -i include/cudnn.h /usr/local/cuda/include 
sudo cp -i lib64/libcudnn* /usr/local/cuda/lib64
sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*
sudo ldconfig /usr/local/cuda/lib64

 

四、采用pip命令安装

（1）安装 pip3 软件包管理器

sudo apt update
sudo apt-get install python3-pip python3-dev  # for Python 3.n
sudo pip3 install --upgrade pip   #更新pip3

（2）在虚拟环境中安装（受推荐的安装方式）

$sudo pip3 install -U virtualenv  # system-wide install
$virtualenv --system-site-packages -p python3 ./venv    # 新建虚拟环境文件夹 venv
$source ./venv/bin/activate  # 激活虚拟环境
(venv) $pip install --upgrade tensorflow      # 若要安装特定版本pip install tensorflow-gpu==1.7

 验证安装并退出虚拟环境：

(venv) $python -c "import tensorflow as tf; tf.enable_eager_execution(); print(tf.reduce_sum(tf.random_normal([1000, 1000])))"
(venv) $deactivate  #退出虚拟环境

 

也可以采用如下代码验证安装（可选）

在 Python 交互式 shell 中输入以下几行简短的程序代码：

# Python
import tensorflow as tf
hello = tf.constant('Hello, TensorFlow!')
sess = tf.Session()
print(sess.run(hello))

如果系统输出以下内容，就说明您可以开始编写 TensorFlow 程序了：

Hello, TensorFlow!

或

b'Hello,TensorFlow!'

五、查看tensorflow版本【可选】

import tensorflow as tf
tf.__version__

卸载（可选）

$ sudo pip3 uninstall tensorflow

 

 

 

故障处理：

(1) ImportError: libcublas.so.9.0: cannot open shared object file: No such file or directory

    CUDA9.0未安装，或安装后路径未加到环境变量。

(2) Loaded runtime CuDNN library: 7103 (compatibility version 7100) but source was compiled with 7005 (compatibility version 7000)

cuDNN版本错误。当前cuDNN的版本是v7.1，应该安装v7.0。

(3)  /sbin/ldconfig.real: /usr/local/cuda/lib64/libcudnn.so.7 is not a symbolic link

在终端输入：

sudo ldconfig -v

在终端中定位到如下语句：

/sbin/ldconfig.real: /usr/local/cuda-9.0/targets/x86_64-linux/lib/libcudnn.so.7 is not a symbolic link

    libcudnn.so.7 -> libcudnn.so.7.3.1

则根据上述输出利用语法 ln -sf source filename，创建软链接：

 sudo ln -sf  /usr/local/cuda/lib64/libcudnn.so.7.3.1  /usr/local/cuda/lib64/libcudnn.so.7