Update Dockerfile
Browse files- Dockerfile +3 -0
Dockerfile
CHANGED
|
@@ -7,6 +7,9 @@ COPY requirements.txt .
|
|
| 7 |
# Install torch FIRST, before flash-attn
|
| 8 |
RUN pip install --no-cache-dir torch
|
| 9 |
|
|
|
|
|
|
|
|
|
|
| 10 |
# Now install all other requirements, including flash-attn
|
| 11 |
RUN pip install --no-cache-dir -r requirements.txt
|
| 12 |
|
|
|
|
| 7 |
# Install torch FIRST, before flash-attn
|
| 8 |
RUN pip install --no-cache-dir torch
|
| 9 |
|
| 10 |
+
# VERIFY torch installation - Add this line
|
| 11 |
+
RUN python -c "import torch; print(torch.__version__)"
|
| 12 |
+
|
| 13 |
# Now install all other requirements, including flash-attn
|
| 14 |
RUN pip install --no-cache-dir -r requirements.txt
|
| 15 |
|