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
|