docker: Error response from daemon: exec: "nvidia-container-runtime-hook": executable file not found in $PATH.

작은소행성☄️ 2022. 1. 22. 14:55

728x90

컨테이너 생성시 gpu 를 설정한 명령어를 사용하면 다음과 같은 에러가 나왔다.

다음 패키지를 설치해서 docker 자체 명령어를 사용해 nvidia-gpu를 컨테이너에서 사용할 수 있도록 지원하는 것으로 보인다.

아래 명령어로 nvidia-container-runtime 을 설치해주면 된다.

apt-get install nvidia-container-runtime

nvidia-smi 로 gpu 확인하고 gpu 를 할당해주면 에러가 생기지 않는것을 확인할 수 있다.

docker run -it --rm --gpus '"device=0,2"' ubuntu nvidia-smi

728x90