Update Dockerfile
Browse files- Dockerfile +1 -0
Dockerfile
CHANGED
|
@@ -61,6 +61,7 @@ RUN echo "Checking NVIDIA packages:" && pip list | grep nvidia || true
|
|
| 61 |
|
| 62 |
# Fix permissions for pip cache
|
| 63 |
RUN mkdir -p /home/user/.cache/pip && chown -R user:user /home/user/.cache
|
|
|
|
| 64 |
|
| 65 |
COPY . .
|
| 66 |
# Install your requirements
|
|
|
|
| 61 |
|
| 62 |
# Fix permissions for pip cache
|
| 63 |
RUN mkdir -p /home/user/.cache/pip && chown -R user:user /home/user/.cache
|
| 64 |
+
RUN python -m xformers.info
|
| 65 |
|
| 66 |
COPY . .
|
| 67 |
# Install your requirements
|