Update Dockerfile

Dockerfile

@@ -1,28 +1,19 @@
 FROM nvidia/cuda:11.3.1-cudnn8-devel-ubuntu18.04
 #CMD nvidia-smi
 
-ENV DEBIAN_FRONTEND=noninteractive \
-    HOME=/home/user \
-    PATH=/home/user/.local/bin:$PATH
+ENV DEBIAN_FRONTEND=noninteractive 
+
+RUN apt-get update -y && apt-get upgrade -y && apt-get install -y sudo && apt-get install -y python3-pip && pip3 install --upgrade pip
+RUN apt-get install -y curl gnupg wget htop sudo git git-lfs software-properties-common build-essential libgl1 zip unzip
 
-# Install Git, Python 3.9, and required packages
-RUN apt-get update && \
-    apt-get install -y software-properties-common && \
-    add-apt-repository ppa:deadsnakes/ppa && \
-    apt-get update && \
-    apt-get install -y git python3.9 python3.9-dev python3-pip wget libgl1-mesa-glx && \
-    apt-get clean && \
-    rm -rf /var/lib/apt/lists/* && \
-    wget https://bootstrap.pypa.io/get-pip.py && \
-    python3.9 get-pip.py && \
-    rm get-pip.py
-    
 # Set up a new user named "user" with user ID 1000
 RUN useradd -m -u 1000 user
 
 # Switch to the "user" user
 USER user
 
+ENV HOME=/home/user \
+    PATH=/home/user/.local/bin:$PATH
 
 # Set the working directory to the user's home directory
 WORKDIR $HOME/app