wangyoucao577 commited on
Commit
504a401
·
verified ·
1 Parent(s): f21bddd
Files changed (1) hide show
  1. Dockerfile +3 -3
Dockerfile CHANGED
@@ -4,9 +4,9 @@ RUN pip install hf_transfer
4
 
5
  # install vllm
6
  #RUN pip install vllm
7
- RUN VLLM_TARGET_DEVICE=cpu
8
- PIP_EXTRA_INDEX_URL=https://download.pytorch.org/whl/cpu \
9
- 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
+ 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