sparkleman commited on
Commit ·
50e995e
1
Parent(s): 353b634
UPDATE
Browse files- Dockerfile +1 -3
Dockerfile
CHANGED
|
@@ -23,15 +23,13 @@ RUN if [ "$MODELSCOPE_ENVIRONMENT" = "studio" ]; then \
|
|
| 23 |
pnpm run build --mode target-rwkv-hf-space; \
|
| 24 |
fi
|
| 25 |
|
| 26 |
-
FROM nvidia/cuda:12.9.0-
|
| 27 |
|
| 28 |
RUN <<EOF
|
| 29 |
apt update
|
| 30 |
apt install --no-install-recommends -y \
|
| 31 |
build-essential \
|
| 32 |
git \
|
| 33 |
-
cuda-nvcc-12-9 \
|
| 34 |
-
cuda-cudart-dev-12-9 \
|
| 35 |
python3-dev \
|
| 36 |
python3-pip \
|
| 37 |
libpython3.10-dev
|
|
|
|
| 23 |
pnpm run build --mode target-rwkv-hf-space; \
|
| 24 |
fi
|
| 25 |
|
| 26 |
+
FROM nvidia/cuda:12.9.0-devel-ubuntu24.04 AS Backend
|
| 27 |
|
| 28 |
RUN <<EOF
|
| 29 |
apt update
|
| 30 |
apt install --no-install-recommends -y \
|
| 31 |
build-essential \
|
| 32 |
git \
|
|
|
|
|
|
|
| 33 |
python3-dev \
|
| 34 |
python3-pip \
|
| 35 |
libpython3.10-dev
|