本文档详细介绍了在新装的Ubuntu18.04系统上，如何通过安装GPU驱动、Docker以及从国内镜像源拉取nvidia-driver和tensorflow镜像，来建立一个能运行TensorFlow的深度学习环境。主要步骤包括更换国内源、安装搜狗输入法、安装Docker、拉取镜像以及在GPU环境下运行TensorFlow。
首先说下,这个是总的环境基于一个新装的崭新的Ubuntu18系统需要以下几个步骤就能跑tensorflow, 不用自己去配置cuda和cudnn;总的就分为以下三步:
第一步 首先你的GPU显卡要先装驱动,就是输入nvidia-smi显示显卡信息就可以了;第二步 安装docker, 具体自己安装 ; 第三步 使用docker pull nvidia-driver 和 tensorflow/tensorflow下面具体tag的image, 然后第一次创建tensorflow容器都要设置nvidia-driver参数设置启动, 第二次之后直接启动容器就行了不用再配置--runtime=nviada了,以下为具体步骤, 如果我没讲清楚, 可以在下面留言:
1.首先安装了用U盘装了Ubuntu18的系统,换国内源, 安装驱动
安装系统之后,设置添加国内镜像源, 会快很多很多!参考百度经验配置, 在如下图配置,不用去手动修改什么source.list 最后保存设置之后, 重启电脑机即可
参考了下面的几位大佬的文章
黄海广：Ubuntu18.04深度学习环境配置(CUDA9+CUDNN7.4+TensorFlow1.8)zhuanlan.zhihu.com
<
最低0.47元/天 解锁文章
762
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
