Ubuntu Anaconda安装TensorFlow-GPU

Ubuntu Anaconda安装TensorFlow-GPU

目前时间是2017.10.22,目前的CUDA最新版本是CUDA9TensorFlow的版本是1.3,一开始安装CUDA9,后来发现TensorFlow不支持,网上有人说可以通过源代码安装TensorFlow,但是我觉得太麻烦了,后来就换了CUDA8,中间遇到了很多版本不对应所产生的问题,下面详细列一下本次安装所使用的安装包的版本号:(亲测可安装成功)

  • CUDA: cuda-repo-ubuntu1604-8-0-local_8.0.44-1_amd64.deb
  • cudnn: cudnn-8.0-linux-x64-v5.1.tar
  • Anaconda: Anaconda3-5.0.0.1-Linux-x86_64.sh

一、安装准备

CUDA下载

cudnn下载

Anaconda下载

在实际的安装过程中,发现目前最新的版本 Tensorflow 并不能支持,后来折腾了好久,才找到上面提到的那几个对应的安装包,百度网盘分享链接,密码:8lpv

二、安装 NVIDIA 驱动

首先打开 terminal,输入一下指令,更新应用列表

sudo apt-get update

多亏了Linux社区的强大支持,我们可以用很简洁的方式安装驱动:

  • 按下win按键,打开菜单

  • 然后在上方的type to search中输入:additional drivers打开 "additional drivers —— 附加驱动",然后选择 与自己显卡匹配的NVIDIA驱动,我这里是 using nvidia binary drivers (375)

  • 然后选择 应用更改,这里安装可能会失败,此时你可以继续选择应用更改,多安装几次即可。

  • 安装完成后,选择restart

三、gcc 降版本

CUDA8.0不支持gcc 5.0及以上的编译器,而系统自带的是5.4及以上版本,因此我们需要降版本,否则会在后面报错。

terminal中输入如下指令,将gcc版本降到 4.9

sudo apt-get install g++-4.9
sudo update-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-4.9 20
sudo update-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-5 10
sudo update-alternatives --install /usr/bin/g++ g++ /usr/bin/g++-4.9 20
sudo update-alternatives --install /usr/bin/g++ g++ /usr/bin/g++-5 10
sudo update-alternatives --install /usr/bin/cc cc /usr/bin/gcc 30
sudo update-alternatives --set cc /usr/bin/gcc
sudo update-alternatives --install /usr/bin/c++ c++ /usr/bin/g++ 30
sudo update-alternatives --set c++ /usr/bin/g++ 

四、安装 CUDA 8.0

进入cuda安装包所在文件夹,右键打开终端,在终端中输入以下代码

# 根据官网上的提示安装 cuda 8.0
sudo dpkg -i cuda-repo-ubuntu1604-8-0-local_8.0.44-1_amd64.deb
sudo apt-get update
sudo apt-get install cuda​

五、安装 cudnn

cudnn-8.0-linux-x64-v5.1.tar解压,右键打开终端,在终端中输入以下代码

sudo cp cuda/include/cudnn.h /usr/local/cuda/include # 复制到 include 中
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64 # 复制到 lib64 中
sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn* # 将头文件复制进去

六、配置CUDA环境变量

开启 GPU 支持:

根据官网教程,我们在terminal中键入下列命令:

sudo gedit ~/.bash_profile # 打开.bash_profile 这是用户的环境变量,不是全局的

然后在打开的文本末尾加入:

export LD_LIBRARY_PATH="$LD_LIBRARY_PATH:/usr/local/cuda/lib64:/usr/local/cuda/extras/CUPTI/lib64"
export CUDA_HOME=/usr/local/cuda

保存并关闭后,输入下列命令使环境变量生效:

source ~/.bash_profile # 使被更改的环境变量生效

安装完成后,一定要通过如下两条命令对驱动进行检验:

  1. NVIDIA的设置界面

    nvidia-settings # 打开 NVIDIA 设置界面
    
NVIDIA设置.png
  1. NVIDIA GPU列表

    nvidia-smi # 这条指令会在 terminal 中产生GPU列表
    
GPU列表.png

网上有些人抄别人的博客,说nvcc -V就可以验证,经过大神实测,存在nvcc -V正常输出但是驱动仍安装失败的现象,因此,上述验证方法是不可信的。

七、安装Anaconda

进入anaconda安装包所在目录,打开终端,根据版本输入安装命令:

bash Anaconda3-5.0.0.1-Linux-x86_64.sh 

回车后,是许可文件,接收许可,anaconda将安装在~/anaconda目录下,直接回车就好。

最后会询问是否把anacondabin添加到用户的环境变量中,选择yes

安装成功以后,在终端输入

source ~/.bashrc

.bashrc中添加的路径生效。

当然,你也可以手动完成环境变量的修改:

  • 首先打开环境变量的文件

    gedit ~/.bashrc
    
  • 然后在文件末尾加入 anaconda3 的路径

    export PATH=/home/你的路径/anaconda3/bin:$PATH
    
  • 最后使我们的改动生效

    source ~/.bashrc
    

这样,我们在terminal中输入python就会默认打开anaconda3里默认的python

八、安装Tensorflow-GPU

创建Tensorflow的运行环境

conda create -n tensorflow python=3.5

激活环境,使用 pip 安装 TensorFlow-GPU

source activate tensorflow
pip install tensorflow-gpu keras # 安装 gpu 版本的 tensorflow 和 keras

安装完成后,我们使用如下命令,即可检验是否成功:

python -c "import keras"

如果看到如下输出,就说明安装成功

安装成功输出.png

参考文章

Ubuntu 安装 tensorflow-gpu + keras

Ubuntu17.04+Nvidia GTX 1080+CUDA 9.0+cuDNN 7.0+TensorFlow 1.3

CUDA 9.0 + cuDNN 7.0 + Tensorflow源码编译

©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 159,015评论 4 362
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 67,262评论 1 292
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 108,727评论 0 243
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 43,986评论 0 205
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 52,363评论 3 287
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 40,610评论 1 219
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 31,871评论 2 312
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 30,582评论 0 198
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 34,297评论 1 242
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 30,551评论 2 246
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 32,053评论 1 260
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 28,385评论 2 253
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 33,035评论 3 236
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 26,079评论 0 8
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 26,841评论 0 195
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 35,648评论 2 274
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 35,550评论 2 270

推荐阅读更多精彩内容