docker-utils

# --------------------------------------------------------------------
# BUILD
# --------------------------------------------------------------------
FROM ubuntu:latest

RUN apt-get update && apt-get upgrade -y
RUN apt-get install curl gnupg2 -y

# --------------------------------------------------------------------
# SETUP GPU Drivers
# --------------------------------------------------------------------
RUN curl -O http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/cuda-repo-ubuntu1804_10.0.130-1_amd64.deb
RUN dpkg -i cuda-repo-ubuntu1804_10.0.130-1_amd64.deb
RUN rm cuda-repo-ubuntu1804_10.0.130-1_amd64.deb
RUN apt-key adv --fetch-keys http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/7fa2af80.pub
RUN apt-get update
RUN DEBIAN_FRONTEND=noninteractive apt-get install cuda -y

# --------------------------------------------------------------------
# SETUP GPU Drivers
# --------------------------------------------------------------------
RUN curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | apt-key add -
RUN curl -s -L https://nvidia.github.io/nvidia-docker/ubuntu18.04/nvidia-docker.list | tee /etc/apt/sources.list.d/nvidia-docker.list

RUN apt-get update
RUN apt-get install nvidia-container-toolkit -y

# docker run --gpus all 417c3cdfead3 nvidia-smi