문서는 Ubuntu 20.04 (WSL), CUDA 12.1을 기준으로 작성되었으며, 다음의 자료(링크, 링크2, 링크3)를 참고한다.
sudo apt update
sudo apt-get install -y ca-certificates curl software-properties-common apt-transport-https gnupg lsb-release
sudo mkdir -p /etc/apt/keyrings
curl -fsSL https://download.docker.com/linux/ubuntu/gpg | sudo gpg --dearmor -o /etc/apt/keyrings/docker.gpg
echo \
"deb [arch=$(dpkg --print-architecture) signed-by=/etc/apt/keyrings/docker.gpg] https://download.docker.com/linux/ubuntu \
$(lsb_release -cs) stable" | sudo tee /etc/apt/sources.list.d/docker.list > /dev/null
sudo apt update
sudo apt install docker-ce docker-ce-cli containerd.io
Docker 컨테이너에서 NVIDIA GPU를 사용하려면 아래의 Toolkit 설치를 추가로 진행해야한다.
distribution=$(. /etc/os-release;echo $ID$VERSION_ID) \
&& curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | sudo apt-key add - \
&& curl -s -L https://nvidia.github.io/nvidia-docker/$distribution/nvidia-docker.list | sudo tee /etc/apt/sources.list.d/nvidia-docker.list
sudo apt-get update
sudo apt-get install -y nvidia-docker2
sudo systemctl restart docker
# WSL에서는 sudo service docker restart 명령어 사용을 권장
sudo docker run --rm --gpus all ubuntu:18.04 nvidia-smi
기본적으로 sudo 권한이 있어야 docker 명령어를 사용할 수 있어 불편하기 때문에 아래의 절차를 추가로 진행하는 것을 권장한다.
sudo groupadd docker
sudo usermod -aG docker $USER
sudo systemctl restart docker
# WSL에서는 sudo service docker restart 명령어 사용을 권장
재부팅
아무 명령어로 테스트
docker images