Added the NVIDIA container toolkit

Dockerfile

@@ -9,4 +9,19 @@ ENV HF_HOME="/app/.cache"
 # DO NOT USE: Depreccated
 # ENV TRANSFORMERS_CACHE="/app/.cache"
 
+RUN apt update \
+	&& apt install -y curl sed
+
+# Add NVIDIA Container Toolkit
+## https://docs.nvidia.com/datacenter/cloud-native/container-toolkit/latest/install-guide.html
+
+RUN curl -fsSL https://nvidia.github.io/libnvidia-container/gpgkey |  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' | \
+     tee /etc/apt/sources.list.d/nvidia-container-toolkit.list \
+
+    && apt update \
+    && apt install -y nvidia-container-toolkit \
+    && rm -rf /var/lib/apt/lists/*
+
 CMD ["f5-tts_infer-gradio", "--port 7860"]