sparkleman commited on
Commit ·
a650820
1
Parent(s): 50e995e
UPDATE
Browse files- Dockerfile +3 -1
Dockerfile
CHANGED
|
@@ -23,13 +23,15 @@ RUN if [ "$MODELSCOPE_ENVIRONMENT" = "studio" ]; then \
|
|
| 23 |
pnpm run build --mode target-rwkv-hf-space; \
|
| 24 |
fi
|
| 25 |
|
| 26 |
-
FROM nvidia/cuda:12.
|
| 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
|
|
|
|
| 23 |
pnpm run build --mode target-rwkv-hf-space; \
|
| 24 |
fi
|
| 25 |
|
| 26 |
+
FROM nvidia/cuda:12.8.1-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 |
+
cuda-nvcc-12-8 \
|
| 34 |
+
cuda-cudart-dev-12-8 \
|
| 35 |
python3-dev \
|
| 36 |
python3-pip \
|
| 37 |
libpython3.10-dev
|