Ubuntu安装TensorFlow

虽然 TensorFlow 官方网站上有比较详细的安装指南，然而我还是磕磕绊绊碰到了许多坑。本文主要记录这些坑，希望对大家有所帮助

1、安装NVIDIA全家桶

注意：从 NVIDIA官网 上下载的CUDA ToolKit中已经包含了相应的驱动，在安装时只需要勾选相应选项就可以完成toolkit和driver的安装。具体步骤如下：

1.1 下载ToolKit。 如果感觉麻烦， 也可以从 我的分享 中下载（9.2版本）链接: https://pan.baidu.com/s/1SiVGkI79XxZ_i7EyO96juw 密码: q7b2

1.2 安装

1.2.1 将linux切换到命令行模式，快捷键 Ctrl + Alt + F2

1.2.2 关闭lightdm服务。 运行

sudo /etc/init.d/lightdm stop

1.2.3 运行安装命令

sudo sh <path/to/you/.sh/file>

1.2.4 勾选需要安装的部件

1.2.5 安装完成运行

nvidia-smi即可查看驱动信息，如未出现则说明安装失败

1.2.6 开启lightdm服务。运行

sudo /etc/init.d/lightdm start

1.3 安装cudnn。下载cudnn有点麻烦，有时登录不上去，因此我也上传到云盘上了。链接: https://pan.baidu.com/s/1CDx1x6vVcbO7c3KqcdrzJw 密码: gn4k

1.3.1 解压压缩包

tar -xf <path/to/your/tar/file>

1.3.2 将对应的文件拷贝至对应文件夹即可

$ 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. /usr/local/cuda/lib64/libcudnn*

1.4 修改环境变量

$ export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/extras/CUPTI/lib64

2、安装依赖环境

2.1 安装bazel 按照该网址安装即可

2.2 安装numpy dev pip wheel

$ sudo apt-get install python-numpy python-dev python-pip python-wheel

3、完成TensorFlow configure

运行源码目录下 ./configure， 按照自己的需求完成配置即可。具体可以参考官方网站

4、安装

4.1 编译pip包

$ bazel build --config=opt --config=cuda //tensorflow/tools/pip_package:build_pip_package

4、使用 pip安装最为简便，一键安装即可。

$ pip install tensorflow # Python 2.7; CPU support (no GPU support)$ pip3 install tensorflow # Python 3.n; CPU support (no GPU support) $ pip install tensorflow-gpu # Python 2.7; GPU support $ pip3 install tensorflow-gpu # Python 3.n; GPU support

如果提示权限不足，请加上 --user选项， 或者使用sudo执行。

5、 如果发现安装的cuda版本与tensorflow不一致，可以参考先卸载NVIDIA驱动 再按 该链接 第二步解决。在此也做一份摘抄，方便查看：

# The 16.04 installer works with 16.10. # download drivers curl -O http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1604/x86_64/cuda-repo-ubuntu1604_9.0.176-1_amd64.deb # download key to allow installation sudo apt-key adv --fetch-keys http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1604/x86_64/7fa2af80.pub # install actual package sudo dpkg -i ./cuda-repo-ubuntu1604_9.0.176-1_amd64.deb # install cuda (but it'll prompt to install other deps, so we try to install twice with a dep update in between sudo apt-get updatesudo apt-get install cuda-9-0