实验室GPU服务器镜像

镜像功能

• 零成本切换cuda版本。（目前支持cu11.1, cu11.7, cu11.8）
• 内置默认使用zsh，安装oh-my-zsh，默认安装spaceship主题，自动安装常用插件，如命令推荐，语法高亮等等。
• 内置安装miniconda，并预先安装好对应cuda版本的pytorch等软件包，无需重复安装。
• 通过与宿主机使用同一网络环境，无需做任何操作即可实现透明代理，访问GitHub等资源。
• 可以灵活配置SSH端口。

创建镜像

选择需要的cuda版本，通过如下命令创建镜像

• -f指定Dockerfile的位置，根据cuda版本指定即可
• -t指定创建镜像的tag
• .以当前所在目录作为docker build的context

cd lab-server-images
docker build --network host -f cuda11.8/Dockerfile -t 0x404/cuda:11.8-beta .

也可以直接从Docker Hub上拉取镜像，使用如下命令拉取cuda 11.8镜像

docker pull 0x404/cuda:cu11.8-beta

创建容器

根据所需要的镜像创建容器，通过如下命令创建容器

• --restart always宿主机重启容器自动重启
• --network host所有容器都直接使用宿主机的网络，方便统一开启代理
• --gpus all指定创建容器所使用的GPU，按需指定即可
• -v test_container_data:/root/workspace，将test_container_data这个volume挂载到容器的/root/workspace，持久化存储
• -e SSH_PORT=2333，使用本仓库提供的镜像必须指定SSH的端口号，才能正确开启SSH服务，该
• -e SE_USER_NAME=test，使用该容器的用户用户名
• -e VOLUME_MNT=workspace，使用本仓库提供的镜像，如果挂载volume，需要设置该环境变量为挂载的位置（默认前缀为/root/，所以这里只需要填workspace即可）

docker run --restart always \
           --network host \
           --gpus all \
           -itd \
           --name test \
           -v test_container_data:/root/workspace \
           -e SSH_PORT=2333 \
           -e SE_USER_NAME=test \
           -e VOLUME_MNT=workspace \
           0x404/cuda:11.7-beta

管理

• portainer