[Docker] gpu 사용 jupyter 컨테이너 띄우기

1. NVIDIA Container Toolkit 설치

• NVIDIA Docker는 이제 NVIDIA Container Toolkit으로 대체되었습니다. 이를 설치하려면, 먼저 NVIDIA의 GPG 키와 저장소를 추가해야 합니다. 다음 명령어를 실행합니다

$ 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

• NVIDIA Container Toolkit을 설치합니다

$ sudo apt-get update
$ sudo apt-get install -y nvidia-docker2

• Docker 서비스를 재시작

$ sudo systemctl restart docker

 

2. Docker jupyter 컨테이너 실행(이미지는 본인이 생성한 이미지 사용)

Docker 컨테이너를 실행하면 됩니다. "--gpus all" 플래그를 사용하여 GPU를 활성화합니다

$ docker run --gpus all -p 8888:8888 jupyter/tensorflow-notebook

 

3. docker 컨테이너 접속

$ docker exec -it jupyter/tensorflow-notebook bash

// nvidia-smi 패키지 설치
$ pip install nvidia-smi
// 그래픽 카드 정보 확인
$ nvidia-smi

주피터 노트북에서 확인 하기