Update Dockerfile
Browse files- Dockerfile +4 -1
Dockerfile
CHANGED
|
@@ -4,10 +4,13 @@ WORKDIR /app
|
|
| 4 |
|
| 5 |
COPY requirements.txt .
|
| 6 |
|
|
|
|
|
|
|
|
|
|
| 7 |
# Install torch FIRST, before flash-attn
|
| 8 |
RUN pip install --no-cache-dir torch
|
| 9 |
|
| 10 |
-
# VERIFY torch installation
|
| 11 |
RUN python -c "import torch; print(torch.__version__)"
|
| 12 |
|
| 13 |
# Now install all other requirements, including flash-attn
|
|
|
|
| 4 |
|
| 5 |
COPY requirements.txt .
|
| 6 |
|
| 7 |
+
# Set environment variable to disable build isolation
|
| 8 |
+
ENV PIP_NO_BUILD_ISOLATION=1
|
| 9 |
+
|
| 10 |
# Install torch FIRST, before flash-attn
|
| 11 |
RUN pip install --no-cache-dir torch
|
| 12 |
|
| 13 |
+
# VERIFY torch installation
|
| 14 |
RUN python -c "import torch; print(torch.__version__)"
|
| 15 |
|
| 16 |
# Now install all other requirements, including flash-attn
|