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

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驱动->应用更改(如图)
NVIDIA驱动

4. 安装nvidia-container-toolkit

4.1 添加nvidia-docker的源

最好还是参考这篇文章https://www.lidaren.com/archives/1950

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

sudo docker run --gpus all -it -p 8888:8888 tensorflow/tensorflow:latest-gpu-jupyter

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

7. 设置用户组

# 这样就不用每次使用docker都需要加sudo了
# create docker group
$ sudo groupadd docker
# add user to the docker group
$ sudo usermod -aG docker $USER
# lougout, change to root
$ sudo su
# log back in, xx is your user name, equals to $USER
$ su xx
# then verify
$ docker run hello-world


版权声明:本文为weixin_44727250原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接和本声明。