首页 > 系统相关> 文章详细

Ubuntu20.04+docker+tensorflow-gpu环境搭建

2021-07-01 21:01:26 阅读：232 来源： 互联网

标签：Ubuntu20.04 TensorFlow sudo nvidia gpu docker 安装 tensorflow

1. Ubuntu20.04双系统的安装

可以看看这个文章：Ubuntu 18.04+Win10双系统安装与配置
关于上面的文章补充几点。

1.1 下载Ubuntu镜像

推荐使用清华大学开源软件镜像站

1.2 确保你的电脑硬盘有空闲的空间（最少20GB）

2. 安装docker

2.1 原因

更好地运行tensorflow框架。Docker 使用容器创建虚拟环境，以便将 TensorFlow 的安装结果与系统的其余部分隔离开来。TensorFlow 程序在此虚拟环境中运行，该环境能够与其主机共享资源（访问目录、使用 GPU、连接到互联网等）。谷歌准备了很多不同版本的TensorFlow Docker images，该映像已经过配置，可运行 TensorFlow。可以为我们省去很多配置的功夫和依赖管理的问题。Docker 容器可在虚拟环境中运行，是设置 GPU 支持的最简单方法。

2.2 安装docker教程

docker官网的教程
亦可参考https://www.cnblogs.com/chengmf/p/13122013.html

3. 安装NVIDIA驱动

打开系统设置->软件和更新->附加驱动->选择NVIDIA驱动->应用更改（如图）

4. 安装nvidia-container-toolkit

4.1 添加nvidia-docker的源

distribution=$(. /etc/os-release;echo $ID$VERSION_ID)
 
curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey|sudo apt-key add -
 
curl -s -L https://nvidia.github.io/nvidia-docker/$distribution/nvidia-docker.list | sudo tee /etc/apt\/sources.list.d/nvidia-docker.lists
 
sudo apt-get update

4.2 安装nvidia-container-toolkit，重启docker

sudo apt-get install -y nvidia-container-toolkit
 
#restart docker
sudo systemctl restart docker

5. tensorflow安装和运行

有了docker，tensorflow的安装和运行也变简单了。见谷歌官网

6. 拉取TensorFlow Docker images

sudo docker pull tensorflow/tensorflow:latest-gpu-jupyter  # latest release w/ GPU support and Jupyter

浏览器访问127.0.0.1:8888下面的链接就可以使用 jupyter notebook：
tensorflow-cmd
在这里插入图片描述环境配置成功，enjoy it!

标签：Ubuntu20.04,TensorFlow,sudo,nvidia,gpu,docker,安装,tensorflow
来源： https://blog.csdn.net/weixin_44727250/article/details/118395104

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

ICode9