Spaces:
Restarting
on
L4
Restarting
on
L4
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
|