[Docker] Docker Setting

 

sudo docker run -it --name test-cuda --gpus all nvidia/cuda:11.2.2-devel-ubuntu20.04 bash

docker start cuda-test
docker attach cuda-test

 

apt update
apt upgrade
apt install python3.8 -y
apt install -y git

apt-get install python3-venv

sudo apt install python3-pip

docker commit test-cuda ubuntu20.04-git-python3.8-cuda11.2

docker tag ubuntu20.04-git-python3.8-cuda11.2 kimjungtaek/ubuntu20.04-git-python3.8-cuda11.2

docker push kimjungtaek/ubuntu20.04-git-python3.8-cuda11.2

 

pip install torch==1.12.1+cu113 torchvision==0.13.1+cu113 torchaudio==0.12.1 --extra-index-url https://download.pytorch.org/whl/cu113

docker commit test-cuda nia-base-torch1.12.1

docker tag nia-base-torch1.12.1 kimjungtaek/nia-base-torch1.12.1

docker push kimjungtaek/nia-base-torch1.12.1