hoonsubin commited on
Commit
18db1d7
·
1 Parent(s): a01fd21

optimize pip install

Browse files
Files changed (1) hide show
  1. 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 .