Added docker commands and gpu_check.py

Dockerfile

@@ -6,6 +6,13 @@ COPY ./requirements.txt /code/requirements.txt
 
 RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
 
+# Install PyTorch for GPU check
+RUN pip install --no-cache-dir torch torchvision
+
+# Add and run the GPU check script
+COPY gpu_check.py /code/gpu_check.py
+RUN python /code/gpu_check.py
+
 # Set up a new user named "user" with user ID 1000
 RUN useradd -m -u 1000 user
 

gpu_check.py

@@ -0,0 +1,7 @@
+import torch
+
+if torch.cuda.is_available():
+    print("CUDA GPU is available")
+    print("GPU:", torch.cuda.get_device_name(0))
+else:
+    print("CUDA GPU is not available")