first

Dockerfile

@@ -4,12 +4,20 @@ ARG DEBIAN_FRONTEND=noninteractive
 
 ENV PYTHONUNBUFFERED=1
 
+RUN curl -fsSL https://nvidia.github.io/libnvidia-container/gpgkey | sudo gpg --dearmor -o /usr/share/keyrings/nvidia-container-toolkit-keyring.gpg \
+  && curl -s -L https://nvidia.github.io/libnvidia-container/stable/deb/nvidia-container-toolkit.list | \
+    sed 's#deb https://#deb [signed-by=/usr/share/keyrings/nvidia-container-toolkit-keyring.gpg] https://#g' | \
+    sudo tee /etc/apt/sources.list.d/nvidia-container-toolkit.list
+
+RUN sed -i -e '/experimental/ s/^#//g' /etc/apt/sources.list.d/nvidia-container-toolkit.list    
+
 RUN apt-get update && apt-get install --no-install-recommends -y \
   build-essential \
   python3.9 \
   python3-pip \
   git \
   ffmpeg \
+  nvidia-container-toolkit \
   && apt-get clean && rm -rf /var/lib/apt/lists/*
 
 WORKDIR /code

requirements.txt

@@ -1 +1,4 @@
+gradio
+transformers
+torch
 einops