Create Dockerfile

fabibo3 · Apr 7, 2022 · 51110ea · 51110ea
1 parent 9e28013
commit 51110ea
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/Dockerfile b/Dockerfile
@@ -0,0 +1,16 @@
+## Pull from existing image
+# FROM nvcr.io/nvidia/pytorch:21.05-py3
+FROM continuumio/miniconda
+
+## Copy requirements
+COPY ./src/requirements.txt .
+
+## Install Python packages in Docker image
+RUN pip3 install -r requirements.txt
+
+## Copy all files
+COPY ./ ./
+
+## Execute the inference command 
+CMD ["./src/run_inference.py"]
+ENTRYPOINT ["python3"]