지식저장소
[Docker] 1. Docker GPU 인식 (Ubuntu)
Magin
2024. 6. 7. 12:51
728x90
1)Docker container내 GPU 인식 확인
#기본적으로 GPU가 인식 되는지 확인
nvidia-smi
위 사진처럼 출력이 된다면 GPU가 인식되고 있는 것 입니다.
만약 위 사진 처럼 출력문이 나오지 않는다면 아래 과정을 따라하면 됩니다.
2) GPG 키 및 APT 키 목록에 추가
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 update
sudo apt install nvidia-docker2 -y
systemctl restart docker
*만약 위 과정 중 도커 버전이 맞지 않아서 설치가 제대로 안될 수도 있다.(버전 확인 필수!!)
이제 컨테이너에 다시 접속하여
"nividia-smi"
명령어를 실행해서 위 사진처럼 나오는지 확인하면 끝!!
+추가사항)
sudo docker run -it --gpus all
이제 컨테이너 생성시 위에 코드처럼 "--gpus all" 옵션을 사용하면 된다.
728x90