기계는 거짓말하지 않는다

Ubuntu Docker daemon 기본 runtime nvidia로 변경 본문

Linux

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 옵션을 사용하여 생성하면 된다.

'Linux' 카테고리의 다른 글

Ubuntu 기본 편집기(editor) 변경  (0) 2024.04.12
Ubuntu netplan static IP 설정(고정 IP 설정)  (0) 2024.03.27
Ubuntu grep -qxF 명령어 활용  (0) 2024.03.15
Docker 이미지 레이어(layer) 병합(Merge), 이미지 압축  (2) 2024.02.29
Linux grep 명령어 -v 옵션 특정 패턴 필터링  (0) 2024.02.25
'Linux' Related Articles more
Comments