wangyoucao577 commited on
Commit
6bf67b1
·
verified ·
1 Parent(s): 504a401

fix: install

Browse files
Files changed (1) hide show
  1. Dockerfile +15 -3
Dockerfile CHANGED
@@ -4,9 +4,21 @@ RUN pip install hf_transfer
4
 
5
  # install vllm
6
  #RUN pip install vllm
7
- ENV PIP_EXTRA_INDEX_URL https://download.pytorch.org/whl/cpu
8
- ENV VLLM_TARGET_DEVICE cpu
9
- RUN pip install https://github.com/vllm-project/vllm
 
 
 
 
 
 
 
 
 
 
 
 
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