ubuntu16.04 安装tensorflow（GPU、CPU）

链接：

https://mirrors.tuna.tsinghua.edu.cn/help/tensorflow/

【注：需要更改成清华源】

GPU安装：

适用GTX1050，GTX1060，GTX1070，GTX1080等高级显卡系列

1. 安装好系统后，更新驱动。系统设置-》软件和更新-》附加驱动-》更新nvidia和intel
2. 安装cuda9

下载 cuda9.run，（deb格式容易出错误）。执行如下命令：

sudo sh cuda_9.0.176_384.81_linux.run

accept

n（不要安装driver）

y

扫描二维码关注公众号，回复： 2361522 查看本文章

y

y

安装完成后，设置环境变量。

设置环境变量：

ctrl+H 打开 (主目录/home/*/) .bashrc文件，添加

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

export LD_LIBRARY_PATH=/usr/local/cuda-9.0/lib64$LD_LIBRARY_PATH　　

##############注意 解决import tensorflow 错误的问题##############　

更改为：（这里运行，不会再出现错误。很抱歉。）

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

终端运行：source ~/.bashrc

检查：nvcc --version

3. 安装cudnn7

下载 for linux版本的，执行如下命令：

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*

4. 安装anaconda（python3.5）

bash Anaconda3-4.2.0-Linux-x86_64.sh

yes（设置环境变量）

5. 安装tensorflow-gpu(任意版本)

复制开始的蓝色链接上的命令。

CPU安装：

直接复制开始链接上的命令安装tensorflow-cpu。