OS 운영체제/Ubuntu

[NVIDIA] 도커 디폴트 런타임 설정 변경

쉬고싶은 거북이 2023. 12. 5. 14:53

Docker default Runtime 설정 변경 방법

Docker의 기본 런타임을 nvidia로 설정해야 docker 빌드 작업 중에 NVCC 컴파일러와 GPU를 사용할 수 있습니다. 

 

/etc/docker/daemon.json 파일 수정

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

 

 

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

    "default-runtime": "nvidia",
}

 

 

 

1) 참조 사이트

https://github.com/dusty-nv/jetson-containers/blob/master/docs/setup.md