updating dockerfile based on chatGPT for error module torch not found

Dockerfile

@@ -6,6 +6,9 @@ COPY ./environment.yml /code/environment.yml
 # Create the environment using the environment.yml file
 RUN conda env create -f /code/environment.yml
 
+# Make RUN commands use the new environment: From ChatGPT
+SHELL ["conda", "run", "-n", "idm", "/bin/bash", "-c"]
+
 # Set up a new user named "user" with user ID 1000
 RUN useradd -m -u 1000 user
 # Switch to the "user" user
@@ -26,4 +29,9 @@ 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
 
-CMD ["./run.sh"]
+# From ChatGPT
+# CMD ["./run.sh"]
+# Make sure conda is initialized for the user
+RUN echo "conda activate idm" >> ~/.bashrc
+
+CMD ["conda", "run", "--no-capture-output", "-n", "idm", "./run.sh"]

environment.yaml

@@ -15,6 +15,7 @@ dependencies:
 
   - pip:
       - accelerate==0.25.0
+      - torch==2.0.1
       - torchmetrics==1.2.1
       - tqdm==4.66.1
       - transformers==4.36.2