FROM nvcr.io/nvidia/cuda:11.6.0-cudnn8-devel-ubuntu20.04

LABEL Arthur="Shankar Mounesh"

RUN apt update && \
    apt install python3-pip -y