快速创建可移植的深度学习Docker环境

0 镜像拉取

首先拉取带有cuda的镜像：nvidia/cuda:11.8.0-cudnn8-runtime-ubuntu18.04
现在镜像被ban了，推荐使用国内镜像站，这里我用的nvcr.io/nvidia/cuda:11.8.0-cudnn8-runtime-ubuntu18.04

docker pull nvcr.io/nvidia/cuda:11.8.0-cudnn8-runtime-ubuntu18.04

当然也可以手动创建镜像，例如ubuntu2004镜像swr.cn-north-4.myhuaweicloud.com/ddn-k8s/docker.io/ubuntu:20.04

1 运行容器

1.1 运行

运行镜像，绑定远程访问的22端口，绑定可外部访问的12346端口

docker run -d -it -p 12225:22 -p 12346:12346 --name 自己起一个名绑定 镜像名

进入镜像

docker exec -it 镜像名 /bin/bash

1.2 换源

sed -i "s@http://.*archive.ubuntu.com@https://mirrors.aliyun.com@g" /etc/apt/sources.list && \
sed -i "s@http://.*security.ubuntu.com@https://mirrors.aliyun.com@g" /etc/apt/sources.list && \
apt update

1.3 安装基本运行包

apt-get update && apt-get install -y \
    git \
    wget \
    vim \
    openssh-server

1.4 下载conda

wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh -O miniconda.sh

直接安装就行，Linux安装conda教程很多

1.5 设置SSH外部访问

下载并修改

apt install -y openssh-server && sed -i 's/#PermitRootLogin prohibit-password/PermitRootLogin yes/' /etc/ssh/sshd_config

开启ssh

service ssh restart

设置ssh密码（可选项，可以不写，也可以写，不强制设置这一步）

echo root:密码 # 密码需要手动替换，例如 echo root:12345

2 发布镜像

将容器创建镜像

docker commit -m "commit" 容器名 仓库名:版本标签

保存镜像到本地

docker save 镜像名/镜像ID -o 镜像保存位置和名字