yusufs commited on
Commit
47b25af
·
verified ·
1 Parent(s): b943191

Update Dockerfile

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