728x90
docker 19.*.* 부터 nvidia-docker가 아닌 nvidia-container-runtime를 설치하면 GPU인식 및 활용이 가능하다.
https://docs.docker.com/config/containers/resource_constraints/#gpu
docker-24.0.7
nvidia-container-runtime 설치
$ sudo apt-get install nvidia-container-runtime
#확인
$ which nvidia-container-runtime-hook
docker restart
재시작 해야 인식된다.
$ sudo systemctl restart docker
GPU 잡히나 확인
docker run -it --rm --gpus all ubuntu nvidia-smi
728x90
'기타 > Docker' 카테고리의 다른 글
도커 이해하기 - 도커 기초 (0) | 2023.12.01 |
---|---|
우분투 도커 관련 명령어 모음집(계속 UPDATE 예정) (0) | 2023.11.29 |
도커 인공지능 환경 구성 - NVIDIA/torch (0) | 2023.11.29 |
도커 입문 - 설치/ Image/ Container (0) | 2023.11.29 |