container 내에서 gpu 사용

Study/docker

soyeonland 2024. 6. 27. 16:22

[docker] docker GPU 설정

예전 글들을 보면 docker container에서 gpu 를 사용하기 위해서는 nvidia-docker를 새로 설치해야된다고 합니다. 하지만, docker 19.*.* 이후부터는 docker 자체에서도 gpu를 지원하게 되었습니다. 따라서 현 시

data-scient2st.tistory.com

1. nvidia-container-runtime 설치

$ apt-get install nvidia-container-runtime

2. 설치가 잘 되었는지 확인

 which nvidia-container-runtime-hook

3. docker restart

재실행하지 않으면, gpu가 인식되지 않음.

sudo systemctl restart docker

docker run -it --gpus all -v 로컬_dir:container_위치 실행할image