NVIDIA Container Toolkit

개요

 NVIDIA社의 GPGPU 가속이 필요한 경우 NVIDIA Docker를 같이 설치해야 한다.

distribution=$(. /etc/os-release;echo $ID$VERSION_ID) && \
curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | apt-key add - && \
curl -s -L https://nvidia.github.io/nvidia-docker/$distribution/nvidia-docker.list | tee /etc/apt/sources.list.d/nvidia-docker.list && \
apt update && apt install -y nvidia-container-toolkit && \
systemctl restart docker

참조 문서

• https://github.com/NVIDIA/nvidia-docker