Vanilla Docker Images

This repository aims to build vanilla Docker images to create desktop versions of docker container (with CUDA, Anaconda3, Pytorch, etc.).

Independencies

docker-ce
docker-compose
nvidia-docker 2.0

Build

Basic Image: Ubuntu18.04 with Anaconda3

You can build the image from the base-image nvidia/cuda with basic softwares installed & some basic settings.

# download the source code
git clone git@github.com:Joy1112/vanilla-docker-images.git
cd vanilla-docker-images

docker build -t <image_name>:<tag> ubuntu-anaconda

# etc.
docker build -t ubuntu18.04-anaconda3/cuda-11.0:latest ubuntu-anaconda

# or u can specify the username & password if u want
docker build -t ubuntu18.04-anaconda3/cuda-11.0:latest \
--build-arg DOCKER_BASE="nvidia/cuda:11.0.3-cudnn8-devel-ubuntu20.04" \
--build-arg USER="dev" \
--build-arg PASSWD="123456" \
ubuntu-anaconda

Softwares have been installed:

vim
wget
curl
openssh-server
git
tmux
htop
Anaconda3 (The sources of pip & conda have been all changed to Tsinghua Open Source)
libgl1-mesa-glx (libGL.so.1)
python3 (/usr/local/bin/python -> /usr/local/bin/python3.6)
iputils-ping
net-tools
iptables
graphviz
tzdata
bzip2
fixuid

Vanilla Pytorch Image

Furthermore, a vanilla pytorch image can be built based on the base-image built above.

docker build -t <image_name>:<tag> vanilla-pytorch

# etc.
docker build -t vanilla-pytorch:1.7.1-cu110-py37 vanilla-pytorch

# or u can adjust the version of python/pytorch/others with argument `--build-arg <var_name>=<var_value>`
docker build -t vanilla-pytorch:1.7.1-cu110-py38 \
--build-arg ENV_NAME="py38" \
--build-arg PYTHON_VERSION="3.8.0" \
vanilla-pytorch

Usage

Command run can be used to launch a new container from the image built and please note the -u argument must be specified.

When u call run command, it will automatically run the script /home/start.sh, which will use fixuid to set the container_user docker's user_id and group_id to your host_user's user_id and group_id and start the ssh service as well.

# arg `-u` must be specified
docker run -it -u $(id -u):$(id -g) <image_name>

# full command
docker run --name <container_name> -it -u $(id -u):$(id -g) -p <new_port>:22 --gpus all -v <path_to_home>/.ssh:/home/docker/.ssh:ro <image_name>

The password for the user docker is also docker.

And user docker have been added into the root group which means you can install any softwares with sudo as the command-prefix.

Future Works

cd $HOME when launch a new container.
fix bug: deal with the losing of other groups when using fixuid. (which is discussed in #30)
use a docker volume to manage the access to the directory /data

Joy1112 / vanilla-docker-images