안녕하세요,
재재입니다.
이번에는 nvidia-docker 설치를 진행해보겠습니다.
(1) package 파일 설치
curl https://get.docker.com | sh \
&& sudo systemctl --now enable docker
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
curl -s -L https://nvidia.github.io/nvidia-container-runtime/experimental/$distribution/nvidia-container-runtime.list | sudo tee /etc/apt/sources.list.d/nvidia-container-runtime.list
sudo apt-get update
(2) nvidia-docker 설치
sudo apt-get install -y nvidia-docker2
이제, 정상적으로 설치가 되었을거에요!
(3) 동작 확인
이제 정상적으로 작동되는지 확인해볼 차례입니다.
sudo nvidia-docker run --rm --gpus all nvidia/cuda:11.0.3-base-ubuntu20.04 nvidia-smi
만약,
docker image 를 성공적으로 가져오지 못하시는 경우에는,
로그인을 먼저 진행해주셔야합니다.
sudo docker login
이때, 사용하시는 계정은 아래 홈페이지에서 등록해주셔야 하는데요.
정상적으로 로그인이 진행되셨다면 다시 입력해주시고,
아래 명령어를 통해 가지고 있는 이미지의 현황을 확인할 수 있습니다.
sudo nvidia-docker images
소스의 출처는 nvidia docs 입니다.
https://docs.nvidia.com/datacenter/cloud-native/container-toolkit/1.6.0/install-guide.html
nvidia-docker 설치하기