Update Dockerfile

Dockerfile

@@ -2,9 +2,10 @@ FROM nvidia/cuda:11.8.0-devel-ubuntu22.04
 # Set up a new user named "user" with user ID 1000
 RUN useradd -m -u 1000 user
 
+# Set up a new user named "user" with user ID 1000
+RUN useradd -m -u 1000 user
 # Switch to the "user" user
-USER root
-
+USER user
 # Set home to the user's home directory
 ENV HOME=/home/user \
 	PATH=/home/user/.local/bin:$PATH
@@ -12,6 +13,9 @@ ENV HOME=/home/user \
 # Set the working directory to the user's home directory
 WORKDIR $HOME/app
 
+# Copy the current directory contents into the container at $HOME/app setting the owner to the user
+COPY --chown=user . $HOME/app
+
 # Install necessary packages
 RUN apt update && apt install git build-essential libopenblas-dev wget python3-pip -y
 RUN apt update && apt install -y wget