=-=
前期准备
注意顺序!!不能先安装python!先装anaconda,再装cuda环境,最后装tensorflow。
设置显卡驱动:右上角系统设置-软件和更新-附加驱动-选则如下设置-应用更改。安完需要重启一下。
安装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。按照下载页的提示安装即可。
配置环境:
$ 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命令来激活/关闭环境辣。