jonatanklosko commited on
Commit
6797ac6
β€’
1 Parent(s): 8ea0b5d

Update base image and remove XLA_TARGET

Browse files
Files changed (1) hide show
  1. Dockerfile +1 -5
Dockerfile CHANGED
@@ -1,4 +1,4 @@
1
- FROM ghcr.io/livebook-dev/livebook:latest-cuda12.1
2
 
3
  ENV LIVEBOOK_APP_SERVICE_NAME="🐳 Hugging Face - $SPACE_TITLE"
4
  ENV LIVEBOOK_APP_SERVICE_URL="https://huggingface.co/spaces/$SPACE_AUTHOR_NAME/$SPACE_REPO_NAME"
@@ -20,9 +20,5 @@ ENV HOME="/home/user"
20
 
21
  USER user
22
 
23
- # TODO: remove this once EXLA v0.8 is out and we infer the default from the environment
24
- # Configure Nx/EXLA to use the GPU (also relevant for warmup_apps below)
25
- ENV XLA_TARGET="cuda120"
26
-
27
  COPY --chown=user public-apps/ /public-apps
28
  RUN /app/bin/warmup_apps
 
1
+ FROM ghcr.io/livebook-dev/livebook:latest-cuda12
2
 
3
  ENV LIVEBOOK_APP_SERVICE_NAME="🐳 Hugging Face - $SPACE_TITLE"
4
  ENV LIVEBOOK_APP_SERVICE_URL="https://huggingface.co/spaces/$SPACE_AUTHOR_NAME/$SPACE_REPO_NAME"
 
20
 
21
  USER user
22
 
 
 
 
 
23
  COPY --chown=user public-apps/ /public-apps
24
  RUN /app/bin/warmup_apps