Spaces:
Build error
Build error
Hoon Kim
commited on
Commit
·
18db1d7
1
Parent(s):
a01fd21
optimize pip install
Browse files- Dockerfile +3 -3
Dockerfile
CHANGED
|
@@ -50,7 +50,7 @@ RUN pip install --upgrade pip setuptools wheel && \
|
|
| 50 |
"cmake>=3.26" \
|
| 51 |
packaging \
|
| 52 |
ninja \
|
| 53 |
-
"setuptools-scm>=8"
|
| 54 |
|
| 55 |
# Install Base PyTorch System - For CPU-only
|
| 56 |
RUN pip install \
|
|
@@ -61,7 +61,7 @@ RUN pip install \
|
|
| 61 |
torchtext \
|
| 62 |
datasets \
|
| 63 |
transformers \
|
| 64 |
-
--extra-index-url https://download.pytorch.org/whl/cpu
|
| 65 |
|
| 66 |
# Install vLLM CPU version from source
|
| 67 |
RUN git clone https://github.com/vllm-project/vllm.git \
|
|
@@ -73,7 +73,7 @@ RUN git clone https://github.com/vllm-project/vllm.git \
|
|
| 73 |
&& rm -rf vllm
|
| 74 |
|
| 75 |
# Fix networkx compatibility
|
| 76 |
-
RUN pip install --force-reinstall networkx==3.2.1
|
| 77 |
|
| 78 |
# Install Python requirements with CPU-only constraints
|
| 79 |
COPY requirements.txt .
|
|
|
|
| 50 |
"cmake>=3.26" \
|
| 51 |
packaging \
|
| 52 |
ninja \
|
| 53 |
+
"setuptools-scm>=8" --no-cache-dir
|
| 54 |
|
| 55 |
# Install Base PyTorch System - For CPU-only
|
| 56 |
RUN pip install \
|
|
|
|
| 61 |
torchtext \
|
| 62 |
datasets \
|
| 63 |
transformers \
|
| 64 |
+
--extra-index-url https://download.pytorch.org/whl/cpu --no-cache-dir
|
| 65 |
|
| 66 |
# Install vLLM CPU version from source
|
| 67 |
RUN git clone https://github.com/vllm-project/vllm.git \
|
|
|
|
| 73 |
&& rm -rf vllm
|
| 74 |
|
| 75 |
# Fix networkx compatibility
|
| 76 |
+
RUN pip install --force-reinstall --no-cache-dir networkx==3.2.1
|
| 77 |
|
| 78 |
# Install Python requirements with CPU-only constraints
|
| 79 |
COPY requirements.txt .
|