Spaces:
Runtime error
Runtime error
Update Dockerfile
Browse files- Dockerfile +1 -1
Dockerfile
CHANGED
|
@@ -1,4 +1,4 @@
|
|
| 1 |
-
|
| 2 |
RUN pip install --nk-cache-dir -r requirements.txt
|
| 3 |
RUN chmod +x /app.py
|
| 4 |
EXPOSE 8000
|
|
|
|
| 1 |
+
FROM pytorch/pytorch:2.0.1-cuda11.7-cudnn8-runtime
|
| 2 |
RUN pip install --nk-cache-dir -r requirements.txt
|
| 3 |
RUN chmod +x /app.py
|
| 4 |
EXPOSE 8000
|