Update Dockerfile

Dockerfile

@@ -1,24 +1,18 @@
 FROM nvidia/cuda:11.3.1-cudnn8-devel-ubuntu18.04
 #CMD nvidia-smi
 
-ENV DEBIAN_FRONTEND noninteractive
+ENV DEBIAN_FRONTEND=noninteractive \
+    HOME=/home/user \
+    PATH=/home/user/.local/bin:$PATH
 
 # Install Git, Python 3.9, and required packages
 RUN apt-get update && \
-    apt-get install -y git python3 python3-dev python3-pip && \
+    apt-get install -y git python3 python3-dev python3-pip wget libgl1-mesa-glx && \
     apt-get clean && \
-    rm -rf /var/lib/apt/lists/*
-
-# Install pip for Python 3.9
-RUN wget https://bootstrap.pypa.io/get-pip.py && \
+    rm -rf /var/lib/apt/lists/* && \
+    wget https://bootstrap.pypa.io/get-pip.py && \
     python get-pip.py && \
     rm get-pip.py
-
-# Install Git
-RUN apt-get update && \
-    apt-get install -y git && \
-    apt-get clean && \
-    rm -rf /var/lib/apt/lists/*
     
 # Set up a new user named "user" with user ID 1000
 RUN useradd -m -u 1000 user
@@ -26,9 +20,6 @@ RUN useradd -m -u 1000 user
 # Switch to the "user" user
 USER user
 
-# Set home to the user's home directory
-ENV HOME=/home/user \
-    PATH=/home/user/.local/bin:$PATH
 
 # Set the working directory to the user's home directory
 WORKDIR $HOME/app