Spaces:
Sleeping
Sleeping
fix: install
Browse files- Dockerfile +15 -3
Dockerfile
CHANGED
|
@@ -4,9 +4,21 @@ RUN pip install hf_transfer
|
|
| 4 |
|
| 5 |
# install vllm
|
| 6 |
#RUN pip install vllm
|
| 7 |
-
|
| 8 |
-
ENV
|
| 9 |
-
RUN pip install
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 10 |
|
| 11 |
# install node and npm for demo
|
| 12 |
ENV NVM_DIR /usr/local/nvm
|
|
|
|
| 4 |
|
| 5 |
# install vllm
|
| 6 |
#RUN pip install vllm
|
| 7 |
+
RUN git clone https://github.com/vllm-project/vllm.git
|
| 8 |
+
#ENV PIP_EXTRA_INDEX_URL https://download.pytorch.org/whl/cpu
|
| 9 |
+
#RUN pip install --upgrade pip
|
| 10 |
+
RUN pip install cmake>=3.26 wheel packaging ninja "setuptools-scm>=8" numpy
|
| 11 |
+
RUN cd vllm && pip install -v -r requirements/cpu.txt --extra-index-url https://download.pytorch.org/whl/cpu
|
| 12 |
+
RUN git clone -b rls-v3.5 https://github.com/oneapi-src/oneDNN.git
|
| 13 |
+
RUN cmake -B ./oneDNN/build -S ./oneDNN -G Ninja -DONEDNN_LIBRARY_TYPE=STATIC \
|
| 14 |
+
-DONEDNN_BUILD_DOC=OFF \
|
| 15 |
+
-DONEDNN_BUILD_EXAMPLES=OFF \
|
| 16 |
+
-DONEDNN_BUILD_TESTS=OFF \
|
| 17 |
+
-DONEDNN_BUILD_GRAPH=OFF \
|
| 18 |
+
-DONEDNN_ENABLE_WORKLOAD=INFERENCE \
|
| 19 |
+
-DONEDNN_ENABLE_PRIMITIVE=MATMUL
|
| 20 |
+
RUN cmake --build ./oneDNN/build --target install --config Release
|
| 21 |
+
RUN cd vllm && VLLM_TARGET_DEVICE=cpu python setup.py install
|
| 22 |
|
| 23 |
# install node and npm for demo
|
| 24 |
ENV NVM_DIR /usr/local/nvm
|