ArtemisDicoTiar / dockers

dockers

Automatically sets environment I prefer on given docker

Requirements

Updated : install Docker and nvidia-driver.

Then follow the instructions in nvidia-docker

Refer to nvidia-docker Usage for parameters [GPUS]

Instructions

cd [DOCKER_YOU_WANT]

docker build -t [IMAGE_NAME] .

docker run --ipc=host -p [HOST_PORT_NAME_FOR_DOCKER]:22 --name=[CONTAINER_NAME] --gpus [GPUS] -v [HOST_DATA_PATH]:/data -v [HOST_CODE_PATH]:/workspace  -dit [IMAGE_NAME]

for example,

cd pytorch

docker build -t torch120 .

docker run --ipc=host -p [HOST_PORT_NAME_FOR_DOCKER]:22 --name=test_torch --gpus 2 -v /hdd/data:/data -v /home/user:/workspace -dit torch120