Spaces:
Paused
Paused
Update Dockerfile
Browse files- Dockerfile +1 -1
Dockerfile
CHANGED
|
@@ -1,7 +1,7 @@
|
|
| 1 |
# ============================================================
|
| 2 |
# GPU-ENABLED SELF-CONTAINED IMAGE FOR DEEPFAKE VIDEO DETECTOR
|
| 3 |
# ============================================================
|
| 4 |
-
FROM
|
| 5 |
ENV DEBIAN_FRONTEND=noninteractive \
|
| 6 |
PYTHONDONTWRITEBYTECODE=1 \
|
| 7 |
PYTHONUNBUFFERED=1
|
|
|
|
| 1 |
# ============================================================
|
| 2 |
# GPU-ENABLED SELF-CONTAINED IMAGE FOR DEEPFAKE VIDEO DETECTOR
|
| 3 |
# ============================================================
|
| 4 |
+
FROM pytorch/pytorch:2.2.2-cuda12.1-cudnn8-runtime
|
| 5 |
ENV DEBIAN_FRONTEND=noninteractive \
|
| 6 |
PYTHONDONTWRITEBYTECODE=1 \
|
| 7 |
PYTHONUNBUFFERED=1
|