Adding deps

Dockerfile

@@ -1,30 +1,37 @@
 FROM continuumio/anaconda3:main
 
-# Set the working directory to /code
-WORKDIR /code
+# Install necessary system libraries
+RUN apt-get update && apt-get install -y \
+    libgl1-mesa-glx \
+    libglib2.0-0
 
-# Copy the environment.yml file into the container
+WORKDIR /code
 COPY ./environment_docker.yml /code/environment_docker.yml
 
 # Create the environment using the environment.yml file
 RUN conda env create -f /code/environment_docker.yml
 
-# Copy the install_deps.sh script into the container
+# Activate the environment and install torch, torchvision, and torchaudio
+RUN /opt/conda/envs/masaenv/bin/conda install -n masaenv -c pytorch -c nvidia -c conda-forge \
+    pytorch==2.1.2 \
+    torchvision==0.16.2 \
+    torchaudio==2.1.2 \
+    pytorch-cuda==11.8 \
+    cudatoolkit==11.8
+
 COPY ./install_deps.sh /code/install_deps.sh
 
 # Make the install_deps.sh script executable
 RUN chmod +x /code/install_deps.sh
 
 # Run the install_deps.sh script
-RUN /code/install_deps.sh
+RUN /bin/bash /code/install_deps.sh
 
 # Set up a new user named "user" with user ID 1000
 RUN useradd -m -u 1000 user
-
 # Switch to the "user" user
 USER user
-
-# Set home to the user's home directory and other environment variables
+# Set home to the user's home directory
 ENV HOME=/home/user \
     PYTHONPATH=$HOME/app \
     PYTHONUNBUFFERED=1 \
@@ -37,11 +44,10 @@ 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 and set the owner to the user
+# Copy the current directory contents into the container at $HOME/app setting the owner to the user
 COPY --chown=user . $HOME/app
 
 # Make run.sh executable
 RUN chmod +x $HOME/app/run.sh
 
-# Command to run the application
 CMD ["./run.sh"]