Update Dockerfile

Dockerfile

@@ -1,8 +1,10 @@
 # Use an official PyTorch image with CUDA support as the base image
 FROM pytorch/pytorch:2.0.1-cuda11.7-cudnn8-runtime
 
-# Install Git
-RUN apt-get update && apt-get install -y git
+# Install Git and system libraries required for OpenGL
+RUN apt-get update && apt-get install -y git \
+    libgl1-mesa-glx \
+    libglib2.0-0
 
 # Set up a new user named "user" with user ID 1000
 RUN useradd -m -u 1000 user