Update Dockerfile

Dockerfile

@@ -1,6 +1,8 @@
 # Use an official PyTorch image with CUDA support as the base image
 FROM pytorch/pytorch:2.0.1-cuda11.7-cudnn8-runtime
 
+RUN apt-get update && apt-get install -y build-essential
+
 # Install a specific version of pip (e.g., 20.2.4)
 RUN pip install pip==20.2.4
 
@@ -13,7 +15,7 @@ RUN apt-get update && apt-get install -y git libgl1-mesa-glx libglib2.0-0
 # Set the pip resolver to "legacy"
 RUN pip config set global.resolver legacy
 
-RUN apt-get update && apt-get install -y ninja-build
+#RUN apt-get update && apt-get install -y ninja-build
 
 # Set up a new user named "user" with user ID 1000
 RUN useradd -m -u 1000 user