tensorflow(2):Ubuntu 16.04安装(GPU版本)

前端之家收集整理的这篇文章主要介绍了tensorflow(2):Ubuntu 16.04安装(GPU版本)前端之家小编觉得挺不错的,现在分享给大家,也给大家做个参考。

其实,机器学习大部分的工作,还是在基于Linux内核的操作系统上完成的,所以再写一篇Ubuntu下的安装过程。
开始步骤其实可以参考我的博客https://blog.csdn.net/ezhchai/article/details/69663082
不论是caffe还是tensorflow,用到GPU都要先安装NVIDIA的驱动和cuda以及cudnn,这个过程没区别。
首先是安装NVIDIA显卡驱动,然后是cuda和cudnn。这些均可以参考caffe安装的博客Caffe开发(1):Ubuntu 16.04环境下caffe配置来实现。
这里需要说明两点:
①安装NVIDIA显卡驱动之前,需要先禁用Ubuntu自带的nouveau驱动,否则会出现显示不正常,以及无法进入系统(Login Loop)等一系列问题。方法如下:

  • 在/etc/modprobe.d/目录下,新建blacklist-nouveau.conf
    sudo vim /etc/modprobe.d/blacklist-nouveau.conf
  • 添加以下语句

    blacklist nouveau options nouveau modset=0 
  • 执行sudo update-initramfs -u,使更改生效;
  • 执行lspci | grep nouveau查看是否有内容,如果没有内容,说明更改生效;如果有内容,就重启一下。
    这里想吐槽一下,Linux做开发挺好,但不太适合办公系统,如果用它来替代windows进行办公的话,对使用者的素质要求会高不少,很多问题不是很容易搞定,严重影响办公效率。所以,如果我们做国产自主系统的话,选择Linux内核做基础,用户友好性上要达到Windows或者MacOS的程度,还有很多工作要做。

②安装cuda和cudnn,要注意版本的匹配。

  • 在下载cudnn时,会提示其对应的cuda版本,cuda和cudnn版本必须匹配;
  • 在安装tensorflow时,要注意tensorflow和cuda以及cudnn版本的匹配,这个在后面的安装过程中会有所体现。

安装完cudnn后,就要关注Python了。Ubuntu16.04自带python2和python3,直接在Terminal中输入python命令,即可进入python环境。如下图所示。

通过python的pip安装tensorflow即可,不过可惜的是pip不是默认自带的软件,需要安装,按提示安装即可。我用python3比较习惯,所以安装python3的pip,也就是pip3。

pip安装完成后,可以直接用pip安装tensorflow。

可以看到安装失败了,网络太差,在线安装下载速度慢,那比较好的方法是直接下载tensorflow的gpu版本的whl包。网址是https://storage.googleapis.com/tensorflow/linux/gpu/tensorflow_gpu-1.3.0-cp35-cp35m-linux_x86_64.whl
之所以选择1.3版本,是因为1.5版本需要cuda9.0,我安装的是8.0,版本不匹配是不能用的。下载好之后,还是用pip安装即可。

可以看到,还是会下一些依赖包,以保证版本匹配。不过这个就小多了,安装成功。这事要测试一下tensorflow是不是能用,简单的方法就是进入python环境,import一下tensorflow,看看成功与否。

很不幸,不成功。这里提示,找不到libcudnn.so.6文件,说明1.3版本要求cudnn版本为6,我的cudnn版本是5,所以需要重新安装cudnn,安装过程也是参考Caffe开发(1):Ubuntu 16.04环境下caffe配置
上图中,可以看到安装好cudnn 6以后,import tensorflow时,就不报错了,说明已经可以使用了。 当然,安装tensorflow有5种方法,还可以用Virtualenv ,Anaconda, Docker和源码安装。有时间会把源码安装过程也写一下,另外几种和pip没什么本质区别,就不尝试了~

猜你在找的Ubuntu相关文章