yusufs commited on
Commit
2f1f5cb
·
verified ·
1 Parent(s): 4e7ea85

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +7 -4
Dockerfile CHANGED
@@ -56,12 +56,15 @@ RUN python3 --version && python3 -m pip --version
56
  # RUN pip install --upgrade pip
57
 
58
  # Install uv, setuptools, and torch for xformers
 
 
 
59
  # Install vLLM
60
- RUN pip install vllm==0.10.0 uv setuptools torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu113
61
 
62
- # Then, install xformers with the --no-build-isolation flag
63
- RUN uv pip install --system \
64
- --no-build-isolation "git+https://github.com/facebookresearch/xformers@v0.0.31"
65
 
66
  RUN useradd -m appuser
67
  USER appuser
 
56
  # RUN pip install --upgrade pip
57
 
58
  # Install uv, setuptools, and torch for xformers
59
+ RUN pip install uv setuptools
60
+
61
+
62
  # Install vLLM
63
+ RUN uv pip install --system vllm==0.10.0 torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu113
64
 
65
+ # # Then, install xformers with the --no-build-isolation flag
66
+ # RUN uv pip install --system \
67
+ # --no-build-isolation "git+https://github.com/facebookresearch/xformers@v0.0.31"
68
 
69
  RUN useradd -m appuser
70
  USER appuser