ubuntu16.04使用anaconda安装gpu版tensorflow + alias为命令创建别名

=-=

前期准备

注意顺序!!不能先安装python!先装anaconda,再装cuda环境,最后装tensorflow。
设置显卡驱动:右上角系统设置-软件和更新-附加驱动-选则如下设置-应用更改。安完需要重启一下。
ubuntu16.04使用anaconda安装gpu版tensorflow + alias为命令创建别名

安装anaconda

我安装的是python 3.5版本的。
先去官网下载。找到linux对应的64bit版本(.sh后缀)
https://www.anaconda.com/download/#linux
下载好mv到自己要安装的地方,终端输入

$ bash Anaconda3-4.2.0-Linux-x86_64.sh

一路回车等待安装好就行啦。

安装CUDA

去官网下载8.0版的,https://developer.nvidia.com/cuda-toolkit。按照下载页的提示安装即可。
ubuntu16.04使用anaconda安装gpu版tensorflow + alias为命令创建别名

配置环境:

$ vim ~/.bashrc

把这段添加进末尾,保存。

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

然后使它生效

$ source ~/.bashrc

安装cudnn

去官网下载,https://developer.nvidia.com/rdp/cudnn-archive。找到对应版本cuDNN v6.0 , for CUDA 8.0,linux版,library,是.tgz后缀。
mv到自己的安装路径下,执行:

$ tar -xzvf cudnn-8.0-linux-x64-v6.0.tgz
$ 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 /usr/local/cuda/lib64/libcudnn*

安装tensorflow

直接用anaconda装就好啦。
创建一个conda环境

$ conda create -n tensorflow-gpu python=3.5

这里如果是python2版本就选2.7,python3就选3.5。这个版本不是自己python的版本,只能从2.7和3.5里选!我选里3.7就失败惹QWQ。
一路yes等他安装。装好后**环境。

$ source activate tensorflow-gpu 

安装tensorflow

$ pip install tensorflow_gpu==1.4

起别名

每次使用都要**环境、关闭环境很麻烦,可以用alias给命令起别名,(像宏一样?
进入~/.barhrc,输入

alias TF='source activate tensorflow-gpu'
alias TFD='source deactivate'

source ~/.bashrc一下。
就可以用TF/TFD命令来**/关闭环境辣。