Add psutil and packaging to build deps for flash-attn
Browse files- Dockerfile +2 -2
Dockerfile
CHANGED
|
@@ -17,8 +17,8 @@ RUN update-alternatives --install /usr/bin/python3 python3 /usr/bin/python3.11 1
|
|
| 17 |
RUN python3 -m venv /opt/venv
|
| 18 |
ENV PATH="/opt/venv/bin:$PATH"
|
| 19 |
|
| 20 |
-
# Upgrade pip
|
| 21 |
-
RUN pip install --upgrade pip wheel setuptools
|
| 22 |
|
| 23 |
# ============================================================================
|
| 24 |
# CRITICAL: UNSLOTH INSTALLATION ORDER
|
|
|
|
| 17 |
RUN python3 -m venv /opt/venv
|
| 18 |
ENV PATH="/opt/venv/bin:$PATH"
|
| 19 |
|
| 20 |
+
# Upgrade pip and install build dependencies
|
| 21 |
+
RUN pip install --upgrade pip wheel setuptools psutil packaging
|
| 22 |
|
| 23 |
# ============================================================================
|
| 24 |
# CRITICAL: UNSLOTH INSTALLATION ORDER
|