새소식

Dev record/Ubuntu

docker: Error response from daemon: could not select device driver "" with capabilities: [[gpu]].

  • -
반응형

Docker에서 GPU를 사용할 때 "could not select device driver "" with capabilities: [[gpu]]" 오류가 발생하는 것은 보통 Docker가 GPU를 인식하지 못해서 발생한다.

  1. NVIDIA Docker 설치 확인:
    • NVIDIA Container Toolkit이 제대로 설치되었는지 확인합니다. 설치되지 않았다면, 다음 명령어로 설치한다.
      sudo apt-get update
      sudo apt-get install -y nvidia-container-toolkit
      sudo systemctl restart docker
  2. Docker와 NVIDIA 드라이버 버전 호환성 확인:
    • Docker 버전과 NVIDIA 드라이버 버전이 호환되는지 확인합니다. 호환성 문제일 수 있음
  3. NVIDIA 드라이버 설치 확인:
    • NVIDIA 드라이버가 제대로 설치되어 있고, nvidia-smi 명령어로 GPU 상태를 확인할 수 있는지 확인
nvidia-smi
  1. Docker 실행 명령어 수정:
    • 위 명령어에서 -gpus all을 사용할 때, NVIDIA Docker가 GPU를 인식할 수 있도록 적절한 권한을 부여해야 한다. -runtime=nvidia 옵션을 추가:
docker run -it --runtime=nvidia --gpus all --ipc=host -v /tmp/.X11-unix:/tmp/.X11-unix -v $HOME/.Xauthority:/home/$(id -un)/.Xauthority -v $HOME/.Xauthority:/root/.Xauthority -v /home/eddie:/home -e NVIDIA_VISIBLE_DEVICES=all -e NVIDIA_DRIVER_CAPABILITIES=all -e DISPLAY=$DISPLAY -e QT_X11_NO_MITSHM=1 --privileged --net=host --name monodepth doyeon0113/monodepth /bin/bash
  1. Docker 및 NVIDIA 관련 로그 확인:
    • 그래도 문제가 해결되지 않는다면, Docker 및 NVIDIA 관련 로그를 확인하여 추가적인 오류 메시지를 확인

docker 재실행

sudo systemctl restart docker

 

정상적으로 작동하는 것을 확인할 수 있다.

반응형
Contents

포스팅 주소를 복사했습니다

이 글이 도움이 되었다면 공감 부탁드립니다.