Up

Dockerfile

@@ -3,7 +3,8 @@
 # FROM ghcr.io/livebook-dev/livebook:edge@sha256:a33c7b012ed367b361d9bc48d0099ca2cc343b150e6c3317589c0224fa351775
 # FROM ghcr.io/livebook-dev/livebook:edge@sha256:11e15be3af529a9ccc47f04f7f0d299867c895cc18dedb6563429b2f4a65c24e
 # FROM ghcr.io/livebook-dev/livebook:edge@sha256:ac80bf4217e1d27d61923211b2c92347d8cae442baad076a28bb86656b5f9594
-FROM ghcr.io/livebook-dev/livebook:edge@sha256:27c5a2e8a932fa61481b9d49b5ab63b2a9819f0439a8d7542a9bc2a73e2749a8
+# FROM ghcr.io/livebook-dev/livebook:edge@sha256:27c5a2e8a932fa61481b9d49b5ab63b2a9819f0439a8d7542a9bc2a73e2749a8
+FROM jonatanklosko/livebook-main:main-ubuntu
 
 ENV LIVEBOOK_APP_SERVICE_NAME "🐳 Hugging Face - $SPACE_TITLE"
 ENV LIVEBOOK_APP_SERVICE_URL "https://huggingface.co/spaces/$SPACE_AUTHOR_NAME/$SPACE_REPO_NAME"
@@ -17,3 +18,9 @@ EXPOSE 7860
 USER root
 RUN mkdir -p /data
 RUN chmod 777 /data
+
+RUN apt update -q && apt install -y ca-certificates wget && \
+    wget -qO /tmp/cuda-keyring.deb https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/cuda-keyring_1.1-1_all.deb && \
+    dpkg -i /tmp/cuda-keyring.deb && apt update -q
+
+RUN apt install -y git cuda-nvcc-12-2 cuda-libraries-12-2 libcudnn8