SParsh003 commited on
Commit
cc0df41
·
verified ·
1 Parent(s): a868e90

Upload Dockerfile with huggingface_hub

Browse files
Files changed (1) hide show
  1. Dockerfile +11 -0
Dockerfile ADDED
@@ -0,0 +1,11 @@
 
 
 
 
 
 
 
 
 
 
 
 
1
+
2
+ FROM pytorch/pytorch:2.2.0-cuda12.1-cudnn8-runtime
3
+
4
+ # Install required system packages
5
+ RUN apt-get update && apt-get install -y git && rm -rf /var/lib/apt/lists/*
6
+
7
+ WORKDIR /app
8
+ COPY runner.py /app/runner.py
9
+ RUN pip install huggingface_hub
10
+
11
+ CMD ["python", "runner.py"]