Spaces:
Paused
Paused
ci(docker): switch to nvidia cuda devel image
Browse files- 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-
|
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
|