ivangabriele commited on
Commit
1b37cec
·
verified ·
1 Parent(s): 7916d6c

ci(docker): switch to cuda image

Browse files
Files changed (1) hide show
  1. Dockerfile +2 -2
Dockerfile CHANGED
@@ -1,6 +1,6 @@
1
  https://huggingface.co/docs/hub/en/spaces-sdks-docker-first-demo#create-the-dockerfile
2
 
3
- FROM python:3.13-bookworm
4
 
5
  ENV RUNNING_IN_DOCKER true
6
 
@@ -26,7 +26,6 @@ RUN chmod 755 /app
26
  USER user
27
  ENV PATH="/home/user/.local/bin:$PATH"
28
  RUN sh -c "$(wget -O- https://github.com/deluan/zsh-in-docker/releases/download/v1.2.1/zsh-in-docker.sh)"
29
- RUN curl -fsSL https://pyenv.run | bash
30
  RUN curl -LsSf https://astral.sh/uv/install.sh | sh
31
 
32
  COPY --chown=user . /app
@@ -34,6 +33,7 @@ COPY --chown=user . /app
34
  RUN ls -la /app
35
 
36
  ENV UV_NO_CACHE="1"
 
37
  RUN uv sync
38
 
39
  SHELL ["/usr/bin/bash", "-c"]
 
1
  https://huggingface.co/docs/hub/en/spaces-sdks-docker-first-demo#create-the-dockerfile
2
 
3
+ FROM nvidia/cuda:12.9.0-cudnn-runtime-ubuntu24.04
4
 
5
  ENV RUNNING_IN_DOCKER true
6
 
 
26
  USER user
27
  ENV PATH="/home/user/.local/bin:$PATH"
28
  RUN sh -c "$(wget -O- https://github.com/deluan/zsh-in-docker/releases/download/v1.2.1/zsh-in-docker.sh)"
 
29
  RUN curl -LsSf https://astral.sh/uv/install.sh | sh
30
 
31
  COPY --chown=user . /app
 
33
  RUN ls -la /app
34
 
35
  ENV UV_NO_CACHE="1"
36
+ RUN uv venv
37
  RUN uv sync
38
 
39
  SHELL ["/usr/bin/bash", "-c"]