Spaces:
Runtime error
Runtime error
Update Dockerfile
Browse files- Dockerfile +12 -2
Dockerfile
CHANGED
@@ -1,7 +1,17 @@
|
|
1 |
FROM nvidia/cuda:12.1.0-runtime-ubuntu20.04
|
2 |
|
3 |
-
# System packages
|
4 |
-
RUN apt-get update && apt-get install -y
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
5 |
|
6 |
# Python packages
|
7 |
RUN pip3 install vllm accelerate
|
|
|
1 |
FROM nvidia/cuda:12.1.0-runtime-ubuntu20.04
|
2 |
|
3 |
+
# System packages (including curl and build tools)
|
4 |
+
RUN apt-get update && apt-get install -y \
|
5 |
+
git \
|
6 |
+
python3 \
|
7 |
+
python3-pip \
|
8 |
+
curl \
|
9 |
+
build-essential \
|
10 |
+
&& rm -rf /var/lib/apt/lists/*
|
11 |
+
|
12 |
+
# Install Rust (needed for tokenizers)
|
13 |
+
RUN curl https://sh.rustup.rs -sSf | sh -s -- -y
|
14 |
+
ENV PATH="/root/.cargo/bin:${PATH}"
|
15 |
|
16 |
# Python packages
|
17 |
RUN pip3 install vllm accelerate
|