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

 

컨테이너 생성시 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

 

 

 

https://docs.docker.com/config/containers/resource_constraints/#gpu