메뉴 건너뛰기

XEDITION

study

[완료] Nvidia Docker (GPU 사용) 설치

proin 2021.02.18 14:29 조회 수 : 1

https://evergrow.tistory.com/13


그냥 일반 Docker 를 사용할 경우 GPU 사용에 제한이 있다.

따라서 그래픽요소가 많이 사용되는 프로그램을 쓰거나, 딥러닝과 같은 연산이 필요한 도커를 실행시킬 경우

GPU를 이어주는 Nvidia-Docker 툴을 설치&사용해야 한다.

 

신규버전이나 사용법은 Nvidia-Docker Github에서 확인하면 된다.

>> https://github.com/NVIDIA/nvidia-docker

 

[ Index ]

#1. 기존 설치내역 삭제(nvidia-docker 버전1 삭제)

#2. Nvidia Docker 패키지 repository 등록

#3. Nvidia-Docker 2 버전 설치 & 데몬 설정

#4. Nvidia Docker 실행 테스트

#5. Nvidia 연동내역 확인

 


 

 

#1. 기존 설치내역 삭제(nvidia-docker 버전1 삭제)

     > nvidia-docker 드라이버 사용 볼륨 제거, nvidia-docker 패키지 전체 삭제

 

 docker volume ls -q -f driver=nvidia-docker | xargs -r -I{} -n1 docker ps -q -a -f volume={} | xargs -r docker rm -f sudo apt-get purge -y nvidia-docker

  *xargs : 앞의 명령어 출력결과를 다음 명령어의 인자(argument)로 사용

 

 

#2. Nvidia Docker 패키지 repository 등록

curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | sudo apt-key add -
distribution=$(. /etc/os-release;echo $ID$VERSION_ID)
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

 

#3. Nvidia-Docker 2 버전 설치 & 데몬 설정

sudo apt-get install -y nvidia-docker2
sudo pkill -SIGHUP dockerd

 

#4. Nvidia Docker 실행 테스트

docker run --runtime=nvidia --rm nvidia/cuda:9.0-base nvidia-smi

 

#5. Nvidia 연동내역 확인

nvidia

docker-ce > nvidia-docker

 

https://hiseon.me/linux/ubuntu/install-docker/

https://cnvrg.io/how-to-setup-docker-and-nvidia-docker-2-0-on-ubuntu-18-04/

 


 

 

 

 

 

번호 제목 글쓴이 날짜 조회 수
104 맥북 프로 M1 빅서 오류 "손상되었기 때문에 열 수 없습니다" 해결 방법 proin 2022.05.02 1
103 YOLOv4, YOLOv4-tiny 분석 proin 2021.12.16 3
102 OpenCV 4.5.1 Linux 설치 proin 2021.09.16 1
101 CUDA, CUDNN 설치 후 nvcc not found가 나올 때 proin 2021.07.15 4
100 %F0%9F%90%B3 Docker * 한글 설정, 한글 깨짐 해결하기 proin 2021.06.16 1
99 Darknet detector 여러가지 오류 처리하기 proin 2021.04.22 1
98 도커(Docker) 컨테이너 로케일 설정 proin 2021.03.31 1
97 리눅스 find -mtime 옵션 개념 익히기 file proin 2021.03.17 3
96 [리눅스/LInux] history 명령어 및 옵션- 이전에 수행한 명령어 리스트 확인, 동작원리 proin 2021.03.15 2
95 리눅스 파일 찾기, 파일속 문자열 찾기 proin 2021.03.03 1
» [완료] Nvidia Docker (GPU 사용) 설치 proin 2021.02.18 1
93 How can I install pip for Python2.7 in Ubuntu 20.04 proin 2021.02.01 1
92 [TF 2.x] TensorFlow 2.0 에서 multi GPU 사용하기 - 텐서플로우 문제 해결 proin 2021.01.26 1
91 Keras, Tensorflow에서 GPU 똑똑하게 사용하기 - 2부 proin 2021.01.26 2
90 Keras, Tensorflow에서 GPU 똑똑하게 사용하기 - 1부 proin 2021.01.26 1
89 Jupyter Notebook 사용법(주피터 노트북 설치 및 사용법) proin 2021.01.25 1
88 Raspberry Pi 3에 Extra Module(contrib) 포함하여 OpenCV 4.5.0 설치하는 방법 proin 2020.12.14 1
87 [Python] 데이터 베이스(mariaDB)를 연결하여 사용하는 방법 proin 2020.11.09 3
86 [Linux] sudo 실행 시 비밀번호 미리 입력하여 실행하기 proin 2020.11.04 1
85 bios 부팅 메뉴에서“Ubuntu”를 어떻게 제거합니까? (UEFI) proin 2020.09.04 1
위로