DegenGamer1702 commited on
Commit
beb8bf8
·
verified ·
1 Parent(s): 7c13787

Update Dockerfile

Browse files
Files changed (1) hide show
  1. Dockerfile +1 -1
Dockerfile CHANGED
@@ -1,7 +1,7 @@
1
  # ============================================================
2
  # GPU-ENABLED SELF-CONTAINED IMAGE FOR DEEPFAKE VIDEO DETECTOR
3
  # ============================================================
4
- FROM python:3.10-slim
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