컨테이너 생성시 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
반응형
'클라우드 및 인프라 > Docker' 카테고리의 다른 글
| [Docker error] docker-compose up error - ERROR: Version in "./docker-compose.yml" is invalid. (0) | 2022.02.19 |
|---|---|
| docker-compose 설치하기 (0) | 2022.02.15 |
| Docker 로 Redis 설치하기 (0) | 2021.12.28 |
| 도커에 mongodb 설치 및 DB 생성하기 (0) | 2021.12.09 |
| 도커 레지스트리에 이미지 올리기 - docker push denied: requested access to the resource is denied (0) | 2021.11.19 |
