이 글은 nvidia-docker를 알고, 컨테이너에서 GPU자원을 사용할 수 있다는 가정하에 설명하고 있습니다.

도커 이미지

TensorRT가 설치된 도커 이미지에 대한 설명은 아래의 사이트에서 얻을 수 있다.
https://docs.nvidia.com/deeplearning/tensorrt/container-release-notes/running.html#running
TensorRT 도커 이미지에는 정말 많은 버전이 있는데 다음의 기준으로 필자는 버전을 선택하고 있다.

• Ubuntu 버전
• Python 버전
• CUDA 버전
• nvidia-driver 버전

위 항목외에도 cuDNN버전과 같이 다른 항목들도 존재하지만, 어떤 우분투 버전, 파이썬 버전, 쿠다 버전을 사용할 것인고 현재 내가 사용하는 nvidia-driver 버전만 알면 이미지 버전을 선택하는데에는 크게 지장이 없다.