docker with tensorflow-gpus

docker 加速

阿里镜像

docker 添加tensorflow 环境

docker pull tensorflow/tensorflow:1.15.4-py3
docker pull tensorflow/tensorflow:1.15.4-gpu-py3
docker exec -it tensorflow-test /bin/bash

docker 下
sudo apt update
sudo apt install libgl1-mesa-glx

安装显卡

ubuntu-drivers devices
== /sys/devices/pci0000:00/0000:00:01.0/0000:01:00.0 ==
modalias : pci:v000010DEd000011C0sv00007377sd00000000bc03sc00i00
vendor : NVIDIA Corporation
model : GK106 [GeForce GTX 660]
driver : nvidia-driver-440-server - distro non-free
driver : nvidia-driver-390 - distro non-free
driver : nvidia-driver-450-server - distro non-free
driver : nvidia-driver-418-server - distro non-free
driver : nvidia-340 - distro non-free
driver : nvidia-driver-450 - distro non-free
driver : nvidia-driver-455 - distro non-free recommended
driver : xserver-xorg-video-nouveau - distro free builtin

sudo ubuntu-drivers autoinstall 就可以自动安装了
or
sudo apt install nvidia-340

docker pull 加速

针对Docker客户端版本大于 1.10.0 的用户

您可以通过修改daemon配置文件/etc/docker/daemon.json来使用加速器

sudo mkdir -p /etc/docker
sudo tee /etc/docker/daemon.json <<-'EOF'
{
  "registry-mirrors": ["https://你的ID.mirror.aliyuncs.com"]
}
EOF
sudo systemctl daemon-reload
sudo systemctl restart docker

运行程序
阿里镜像站

Docker apt-get速度很慢，设置docker ubuntu apt-get 源

LInux下使用docker镜像（tensorflow-gpu）

Linux下docker容器的打包

升级docker19.03使用nvidia-container-toolkit

sudo modprobe 8821cu

nvidia docker