Ubuntu 18.04配置tensorflow-gpu 加 cuda9.1 加cudnn7.05

  以前一直觉得配置环境挺简单的,但是自从这次新买电脑开始配置gpu版之后发现配置个gpu版的tensorflow是真的不省心。下面记录一下我的安装历程。
首先尝试了下在windows上安装,三天之后,终于以失败收场,看来我的水平是用不了windows了,于是选择了安装linux双系统,在linux上尝试了一下午,终于搞定了。
  现在将我的安装过程记录下来,希望给安装同样配置的同学一些提示:

  • 首先是安装linux双系统,在这里选择了安装最新的ubuntu18.04

首先和安装windows一样,先制作一个U盘启动盘,最终选择的u盘制作工具为universal usb installer(据说这个工具很好用,尝试了之后发现确实很好用)
附上下载链接:https://universal-usb-installer.en.softonic.com/
是免费使用的,先下载镜像,然后按照步骤装就行了,这步没什么出错的机会,分区什么的我都是按照的默认的去装的。

  • 然后装cuda9.1

感觉这是个大坑,后来者还是装9.0的比较好,或者装最新的9.2也行,我网上查的上一个教程是装9.1的,装完之后只能装到tf1.8.0,最新的1.9.0只有用cuda9.2才能装上,或者装9.0直接就能pip install tensorflow-gpu装,不用找github上专门为9.1/9.2编译的tensorflow。
从官网下载就很快,我这里速度是5m/s左右吧,四个文件都要下载(教程上说的,我没有尝试只下载主程序行不行)参考这个教程:https://blog.csdn.net/breeze5428/article/details/78256915/
注意,配置/.bashrc文件很重要,我一直是这一步卡了好久,一直出现找不到文件的错误(具体什么错误记不得了)。
安装完成后,测试cuda是否成功,nvidia-smi
然后装cudnn,注意版本一定要匹配,可以上官网上看看跟什么版本搭(这步我没有出错。)然后把文件复制或者移动到cuda对应文件夹下就ok了。

  • 最后是tensorflow的安装

这步挺难受的,只能安装github里的版本,链接如下:https://github.com/mind/wheels
选好版本下载whl文件后,pip安装就ok。
第一次写安装过程,坑都提及到了,但是没说清楚,下次安装的时候我注意截个图,把过程在描述的详细点,这次先就这样了。

你可能感兴趣的:(Ubuntu 18.04配置tensorflow-gpu 加 cuda9.1 加cudnn7.05)