Hugging Face's logo

Spaces:
AJMC2002
/
Neural-Networks
Sleeping

App Files Community
AJMC2002 commited on
Commit
f76629d
·
verified ·
1 Parent(s): 37e3a8e

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +3 -1
Dockerfile CHANGED
@@ -1,4 +1,4 @@
1
- FROM ghcr.io/livebook-dev/livebook:latest-cuda11.8
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,6 +20,8 @@ ENV HOME="/home/user"
20
 
21
  USER user
22
 
 
 
23
  COPY --chown=user public-apps/ /public-apps
24
  COPY --chown=user plates/ /home/user/plates
25
 
 
1
+ FROM ghcr.io/livebook-dev/livebook:latest-cuda
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
+ ENV XLA_TARGET="cuda120"
24
+
25
  COPY --chown=user public-apps/ /public-apps
26
  COPY --chown=user plates/ /home/user/plates
27