Adding deps

Dockerfile

@@ -11,14 +11,6 @@ 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
 
-# Activate the environment and install torch, torchvision, and torchaudio
-RUN /bin/bash -c "source activate masaenv && 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

install_deps.sh

@@ -3,6 +3,10 @@
 # Exit script if any command fails
 set -e
 
+#Install torch
+
+pip install torch torchvision torchaudio
+
 # Update pip and install openmim
 echo "Updating pip and installing openmim..."
 pip install -U pip