ivangabriele commited on
Commit
1f3268c
·
verified ·
1 Parent(s): 50d0b86

ci(docker): switch to nvidia cuda devel image

Browse files
Files changed (1) hide show
  1. Dockerfile +3 -1
Dockerfile CHANGED
@@ -1,6 +1,6 @@
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
 
@@ -35,6 +35,8 @@ RUN uv sync
35
 
36
  SHELL ["/usr/bin/bash", "-c"]
37
 
 
 
38
  RUN source .venv/bin/activate
39
 
40
  # `7860` is the default port for Hugging Face Spaces running on Docker
 
1
  # https://huggingface.co/docs/hub/en/spaces-sdks-docker-first-demo#create-the-dockerfile
2
 
3
+ FROM nvidia/cuda:12.9.0-cudnn-devel-ubuntu24.04
4
 
5
  ENV RUNNING_IN_DOCKER=true
6
 
 
35
 
36
  SHELL ["/usr/bin/bash", "-c"]
37
 
38
+ # ENV CUDA_HOME="/usr/local/cuda"
39
+
40
  RUN source .venv/bin/activate
41
 
42
  # `7860` is the default port for Hugging Face Spaces running on Docker