Create Dockerfile

Dockerfile

@@ -0,0 +1,17 @@
+# Use an official PyTorch runtime as parent image
+FROM pytorch/pytorch:1.10.0-cuda11.1-cudnn8-runtime
+
+# Set the working directory in the container
+WORKDIR /app
+
+# Copy the current directory contents into the container at /app
+COPY . /app
+
+# Install any needed packages specified in requirements.txt
+RUN pip install --no-cache-dir -r requirements.txt
+
+# Define environment variable
+ENV NAME World
+
+# Run app.py when the container launches
+CMD ["python", "app.py"]