第二步
安装Anaconda(自带Python3.6)
通过网站:https://www.continuum.io/downloads 下载Anaconda,本人下载的是:Anaconda3-4.3.1-Linux-x86_64.sh
执行命令:
bashAnaconda3-4.3.1-Linux-x86_64.sh
安装过程中,基本都是选择yes,安装成功后,会有当前用户根目录下生成一个anaconda3的文件夹,里面就是安装好的内容
第三步 安装GPU驱动
1.首先查看自己的GPU显卡版本
执行命令:
本人的显卡信息如下图
2.在http://www.nvidia.com/Download/index.aspx官网上下载相对应版本的显卡驱动
3.禁止默认显卡驱动
打开blacklist.conf文件:
执行命令:
sudo gedit /etc/modprobe.d/blacklist.conf
在里面添加以下内容:
4.安装 kernel-devel
在http://kernel.ubuntu.com/~kernel-ppa/mainline/v4.8/官网上下载三个文件(注意:根据自己系统的版本),如下图
然后安装
执行命令:
sudo dpkg -i linux-*.deb
5.安装显卡驱动
首先关闭X-Window
执行命令:
sudo service lightdm stop
然后重启系统,通过Ctrl+Alt+F1进入init3(文本模式),登路账户和密码进行显卡驱动安装
执行命令:
chmod +x NVIDIA-Linux-x86_64-375.51.run
sudo ./NVIDIA-Linux-x86_64-375.51.run
经过一系列的接受和同意,就可以啦
接着开启X-Window
执行命令:
sudo service lightdm start
另外,可通过下面命令查看GPU信息
执行命令:
sudo nvidia-xconfig -query-gpu-info
第四步 安装CUDA
1.首先在https://developer.nvidia.com/cuda-downloads官网上下载
本人下载的是CUDA 8.0
2.然后安装
执行命令:
chmod +x cuda_8.0.61_375.26_linux.run
sudo ./cuda_8.0.61_375.26_linux.run
过程一系列的接受和同意就可以。(注意:若缺少什么工具,可根据对应的提示进行安装)
3.添加环境变量
执行命令:
添加内容如下:
PATH=/usr/local/cuda-8.0/bin:$PATH
export PATH
export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64:/lib
保存并退出
然后使之立刻生效
执行命令:
4.添加lib库路径
执行命令:
sudo vi /etc/ld.so.conf.d/cuda.conf
添加如下内容:
/usr/local/cuda-8.0/lib64
保存并退出
然后使之立刻生效
5.验证安装
执行命令:
本人服务器信息如下图:
第五步 安装cuDNN
首先在https://developer.nvidia.com/cudnn官网上下载(注意版本,必须是5.1)
解压
然后依次执行以下命令:
cd cuda
sudo cp lib64/lib* /usr/local/cuda/lib64/
sudo cp include/cudnn.h /usr/local/cuda/include/
更新链接
cd /usr/local/cuda/lib64/
sudo rm -rf libcudnn.so libcudnn.so.5
sudo ln -s libcudnn.so.5.1.10 libcudnn.so.5
sudo ln -s libcudnn.so.5 libcudnn.so
第六步 安装TensorFlow
终于快结束了。。。。。。
1.首先在官网上下载GPU版TensorFlow(注意:要与Python版本相对应)
然后执行如下命令:
sudo pip install tensorflow_gpu-1.1.0rc2-cp36-cp36m-manylinux1_x86_64.whl#md5\=8
37b77c371fc7c04a27923e8f3eda2a6.whl
过程如下图:
2.测试TensorFlow
执行命令:
import tensorflow as tf
引包tensorflow包,如果没有报错,则安装成功,否则就有问题。
然后输入:
tf.__version__
tf.__path__
可查看tensorflow的安装版本和安装路径(左右各两根下横线)
最后总结:心累,还是心累。有些坑,一个人摸索了很久才走出来。特此记录,仅供参考,若有问题,敬请指正。之前一直用的是Centos系统上的TensorFlow,和在Ubuntu系统上相比,安装过程还是有些不同。Good luck!