ubuntu20.04+3090+docker+tf1.x全套安装流程

UBUNTU 20上的docker安装tf1.x

1 在 Ubuntu 20.04 上安装 Docker

在 Ubuntu 上安装 Docker 非常直接。我们将会启用 Docker 软件源，导入 GPG key，并且安装软件包。

首先，更新软件包索引，并且安装必要的依赖软件，来添加一个新的 HTTPS 软件源：

sudo apt update
sudo apt install apt-transport-https ca-certificates curl gnupg-agent software-properties-common

使用下面的 curl 导入源仓库的 GPG key：

curl -fsSL https://download.docker.com/linux/ubuntu/gpg | sudo apt-key add -

将 Docker APT 软件源添加到你的系统：

sudo add-apt-repository "deb [arch=amd64] https://download.docker.com/linux/ubuntu $(lsb_release -cs) stable"

现在，Docker 软件源被启用了，你可以安装软件源中任何可用的 Docker 版本。

1.1 安装docker

安装最新的docker

sudo apt update
sudo apt install docker-ce docker-ce-cli containerd.io

1.2 以非 Root 用户身份执行 Docker

创建docker组

$ sudo groupadd docker

将当前用户加入到docker

$ sudo usermod -aG docker $USER

切换到docker群组

$ newgrp docker

不使用root用户执行下面的命令

$ docker run hello-world

如果还报错，可以尝试重启。

$ reboot

报下面的信息，说明执行成功了。

2 docker上安装tf1.5-gpu

2.1 安装NVIDIA Driver

去英伟达官网下载驱动，已经下载成功了，忽略这个步骤。我选择的版本如下。

点击下载，进行安装。

2.2 docker上安装tf1.x

docker pull tensorflow/tensorflow:1.15.5-gpu


docker container run -it --runtime=nvidia tensorflow/tensorflow:1.15.5-gpu bash

2.2.1 接下来解决报错

docker: Error response from daemon: Unknown runtime specified nvidia.

这个时候是我们需要安装nvidia-docker2

sudo apt-get install -y nvidia-docker2

执行安装nvidia-docker2步骤时：

root@docker:~# sudo apt-get install nvidia-docker2
Reading package lists... Done
Building dependency tree       
Reading state information... Done
E: Unable to locate package nvidia-docker2

完整执行命令如下：

curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | \
sudo apt-key add -

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

# 安装nvidia-docker2软件包并重新加载docker守护程序配置
sudo apt-get install -y nvidia-docker2
sudo pkill -SIGHUP dockerd

3 验证docker中的tf是否安装成功

3.1 启动docker

docker container run -it --runtime=nvidia tensorflow/tensorflow:1.15.5-gpu bash

3.2 开始验证

输入python，进入命令行

import tensorflow as tf

print(tf.version)

print(tf.test.is_gpu_available())

版本号是1.15.5，gpu状态是True，说明安装成功。