soyeonland

container 내에서 gpu 사용 본문

Study/docker

container 내에서 gpu 사용

soyeonland 2024. 6. 27. 16:22

https://data-scient2st.tistory.com/46

 

[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

'Study > docker' 카테고리의 다른 글

docker container 안에서의 변화 저장  (0) 2024.08.20
local directory 에 docker mount  (0) 2024.06.27
docker saving credinetials error  (0) 2024.06.27
tensorflow docker use  (0) 2024.06.27
docker none 지우기  (0) 2024.06.27