[docker] nvidia 연결 끊겼을 때

상황

docker에서 yolo 학습 돌리는데 갑자기 cpu로 학습이 돌아갔다

nvidia-smi를 검색해봤는데 에러가 났다

 

 

해결방법

vi /etc/docker/daemon.json

 

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

$ service docker restart