设为首页 加入收藏

TOP

Ubuntu16.04.1如何安装TensorFlow1.1.0(GPU版)
2019-05-11 00:29:46 】 浏览:44
Tags:Ubuntu16.04.1 如何 安装 TensorFlow1.1.0 GPU
第一步 更新最新的软件包信息
执行命令:
sudo apt-get update
第二步 安装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显卡版本
执行命令:
lspci
本人的显卡信息如下图

2.http://www.nvidia.com/Download/index.aspx官网上下载相对应版本的显卡驱动

3.禁止默认显卡驱动
打开blacklist.conf文件:
执行命令:
sudo gedit /etc/modprobe.d/blacklist.conf
在里面添加以下内容:
blacklist nouveau
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.添加环境变量
执行命令:
sudo vi /etc/profile
添加内容如下:
PATH=/usr/local/cuda-8.0/bin:$PATH 
export PATH
export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64:/lib
保存并退出
然后使之立刻生效
执行命令:
source /etc/profile
4.添加lib库路径
执行命令:
sudo vi /etc/ld.so.conf.d/cuda.conf
添加如下内容:
/usr/local/cuda-8.0/lib64
保存并退出
然后使之立刻生效
执行命令:
sudo ldconfig
5.验证安装
执行命令:
nvcc -V
本人服务器信息如下图:


第五步 安装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
在终端输入python,进入python编译环境
执行命令:
import tensorflow as tf
引包tensorflow包,如果没有报错,则安装成功,否则就有问题。
然后输入:
tf.__version__
tf.__path__
可查看tensorflow的安装版本和安装路径(左右各两根下横线)

最后总结:心累,还是心累。有些坑,一个人摸索了很久才走出来。特此记录,仅供参考,若有问题,敬请指正。之前一直用的是Centos系统上的TensorFlow,和在Ubuntu系统上相比,安装过程还是有些不同。Good luck!
】【打印繁体】【投稿】【收藏】 【推荐】【举报】【评论】 【关闭】 【返回顶部
上一篇我的第一个页面 下一篇设计一个LRU Cache

最新文章

热门文章

Hot 文章

Python

C 语言

C++基础

大数据基础

linux编程基础

C/C++面试题目