반응형
docker-compose에서 GPU 사용하기
-
docker-compose 에서 GPU 사용하기OS/Linux 2022. 5. 27. 23:49
docker 에서 GPU 사용시 꼭 nvidia에서 제공하는 toolkit 설치해야 합니다. 그리고 해당 서버에 nvidia driver가 설치되어 있어야 합니다. docker run 할 때와 docker-compose up 할 때의 셋팅법이 다르니 꼭 compose 까지 셋팅을 해야합니다. 1. nvidia toolkit install Installation Guide — NVIDIA Cloud Native Technologies documentation 2. docker-compose.yml 수정 Enabling GPU access with Compose runtime: nvidia 추가 (test) services: test: image: nvidia/cuda:10.2-base command: nv..