update dockerfile

docker/Dockerfile

@@ -1,7 +1,5 @@
-FROM tensorflow/tensorflow:2.9.0-gpu
-
-RUN apt-get update
-RUN apt-get upgrade -y
+# Use an existing image as the base
+FROM nvcr.io/nvidia/pytorch:23.03-py3
 
 # Add the current user to the image
 ARG USER_NAME=lhunor
@@ -10,7 +8,7 @@ ARG GROUP_NAME=guest
 ARG GROUP_ID=11000
 
 # Update the system and install required packages
-RUN apt-get update && apt-get upgrade -y && apt-get install -y sudo git
+RUN apt-get update && apt-get upgrade -y && apt-get install -y sudo git apt-utils
 
 # Create a new group with the same GID as the host
 RUN groupadd --gid $GROUP_ID $GROUP_NAME
@@ -28,10 +26,12 @@ RUN usermod -aG sudo $USER_NAME
 # Set the default user to the new user
 USER $USER_NAME
 
-WORKDIR //home-net
+WORKDIR /workspace
 
-RUN apt-get install git ffmpeg libsm6 libxext6 libopenexr-dev zip unzip htop nano -y
-RUN pip install numpy matplotlib opencv-python wandb OpenEXR tensorflow-graphics scipy plotly pandas tensorboardX opencv-python cmapy
+ENV PATH="$PATH:/home/$USER_NAME/.local/bin"
+RUN pip install --upgrade pip
+COPY requirements.txt /tmp/
+RUN pip install --requirement /tmp/requirements.txt
 
 # defining the volumes that will be mounted
 VOLUME [ "/data-net" ]