[Docker] 1. Docker GPU 인식 (Ubuntu)

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" 옵션을 사용하면 된다.