Ubuntu16.04环境下tensorflow_gpu版安装

        苦逼的大学四年过去了,又到了毕业就失业的时候。心里迷茫有惆怅,简历到处瞎投,大公司进不去,蛇皮公司不想去。机缘巧合下进了一家搞人工智能的公司,于是开始学习一下一些深度学习的东西。熟悉一下tensorflow框架,看看能不能做一点东西。windows+cpu版的tensorflow网上教程一遍下来毫无错误,这个就不讲了。说一说我装gpu版的tensorflow的艰苦历程(因为gpu的计算速度比cpu快,一般企业都使用gpu来进行模型的训练)。

        因为tensorflow更新得很快,而且gpu版本的tensorflow是必须要cuda、cudnn的支持的,所以他们的版本一定要匹配,千万不要全下载最新版的,这样会出现很多的问题,比如缺失libcudnn.so文件等等,我安装的是tensorflow1.4 + cuda8.0 +cuDNNv6 for CUDA8.0  这样的组合,亲测是可用。

1. 安装ubuntu16.04

        安装ubuntu16.04单系统的过程也没什么难度,安装wondows+Ubuntu双系统会涉及到压缩分区等一些操作,有一些麻烦。单系统的安装具体可以参照百度经验这个安装:点击打开链接

2. 安装显卡驱动

        按Ctrl+Alt+T调出终端,在终端里输入

  nvidia-smi

        出现以下类似界面,则你的显卡驱动已经安装好了,可以进行下一步,否则就需要带NVDIA官网找到你合适的驱动进行安装。

若是显示命令不存在则需要到NVDIA官网上去下载合适的版本的驱动    http://www.geforce.cn/drivers

也可以参照这篇文章  https://blog.csdn.net/tianrolin/article/details/52830422

3.安装cuda8.0

     去官网默认的是cuda9.2,找来找去都找不到cuda8.0在哪,直接给大家一个链接  点进去就行了  https://developer.nvidia.com/cuda-80-ga2-download-archive


根据自己电脑的配置来选择就好了,下载runfile文件,然后cd到下载好的文件的文件夹下,输入

sudo sh cuda_8.0.61_375.26_linux.run

这样就搞定了。

4.安装cudnn v6 for cuda8.0

        到官网上去下载cuDNNv6  链接在此  https://developer.nvidia.com/rdp/cudnn-archive

下载cuDNN v6.0 Library for Linux ,然后使用cd 命令跳到下载目录下,或者使用浏览器的打开下载目录,右键空白处-->选择 “从终端打开”,然后执行以下命令

sudo tar -xzvf cudnn-8.0-linux-x64-v5.1.tgz   //此处输入sudo tar -xzvf cudnn-后按Tab键就会自动补全了
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*

这样cudnn就搞定了。

6.Tensorflow_gpu1.4版本的安装

    使用pip工具安装tensorflowgpu非常方便,使用pip工具之前需要先进行pip的安装

sudo apt install python-pip //安装Python2.x版本的pip

sudo apt install python-pip3 //安装Python3.x版本的pip3

如果使用pip进行tensorflow_gpu的安装的话,就必须使用python2.x进行编译(因为pip是将tensorflow包安装到python2的目录下,而pip3是将tensorflow安装到python3的目录下),同理使用Python3.x版本Python引入tensorflow就必须使用pip3进行tensorflow的安装。

如果你使用的是python2.x版本,执行以下指令

sudo pip install tensorflow_gpu == 1.4
如果你使用的是python3.x版本,执行以下指令
sudo pip3 install tensorflow_gpu == 1.4

输入管理员密码就好了

7.测试

  因为tensorflow需要用到numpy、matplotlib包这样依赖也可以通过pip或者pip3进行安装。

>>python3
>>import tensorflow as tf
>>x = tf.constant(0.1)
>>y = tf.constant(0.2)
>>s = tf.Session()
>>print(x+y)
>>print(s.run(x+y))


好了,这样就万事大吉了。这是我安装好后按照回忆写的,如果有不对的地方请大家多多指正~~感谢阅读,希望能帮到你。



评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值