Add Dockerfile

Dockerfile

+# Use the official Ubuntu base image
+FROM ubuntu:22.04
+
+# Set environment variables
+ENV DEBIAN_FRONTEND=noninteractive
+
+# Install Python and pip
+RUN apt-get update && apt-get install -y \
+    python3 \
+    python3-pip && \
+    rm -rf /var/lib/apt/lists/*
+
+# Install PyTorch
+RUN pip3 install torch
+
+# Command to check the number of GPUs available
+CMD ["python3", "-c", "import torch; print('GPUs available:', torch.cuda.device_count())"]
\ No newline at end of file