기계는 거짓말하지 않는다

Ubuntu Docker daemon 기본 runtime nvidia 변경 본문

Container Tools/Docker

Ubuntu Docker daemon 기본 runtime nvidia 변경

KillinTime 2024. 3. 22. 23:49

Docker container에서 NVIDIA GPU를 사용하려면

NVIDIA container toolkit을 설치하고 runtime을 nvidia로 설정하여야 한다.

Docker daemon.json 파일은 /etc/docker/daemon.json 경로에 있거나 없다면 추가한다.

deamon.json 수정 또는 생성

sudo vi /etc/docker/daemon.json

daemon.json 파일에 다음 내용을 추가한다.

{
    "default-runtime": "nvidia",
    "runtimes": {
        "nvidia": {
            "path": "nvidia-container-runtime",
            "runtimeArgs": []
        }
    }
}

Docker 재시작

sudo service docker restart
# 또는
sudo systemctl restart docker

컨테이너 생성 시 --gpus 옵션을 사용하여 생성하면 된다.

'Container Tools > Docker' 카테고리의 다른 글

Ubuntu Docker 이미지 저장 위치 변경  (0) 2024.08.27
Docker 컨테이너 내부 한글 locale UTF-8 환경 변수 설정  (0) 2024.07.14
Docker 이미지 레이어(layer) 병합(Merge), 이미지 압축  (2) 2024.02.29
빈 도커 이미지(empty docker image) 생성  (0) 2024.02.25
Docker 컨테이너 이미지로 저장, 이미지 백업, 이미지 불러오기 (commit, save, load)  (0) 2023.02.28
'Container Tools/Docker' Related Articles more
Comments