도커 GPU 설정하기 Nvidia Toolkit

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