JohanDL commited on
Commit
feb061d
1 Parent(s): ce35c7e

Adding deps

Browse files
Files changed (1) hide show
  1. Dockerfile +2 -2
Dockerfile CHANGED
@@ -12,12 +12,12 @@ COPY ./environment_docker.yml /code/environment_docker.yml
12
  RUN conda env create -f /code/environment_docker.yml
13
 
14
  # Activate the environment and install torch, torchvision, and torchaudio
15
- RUN /opt/conda/envs/masaenv/bin/conda install -n masaenv -c pytorch -c nvidia -c conda-forge \
16
  pytorch==2.1.2 \
17
  torchvision==0.16.2 \
18
  torchaudio==2.1.2 \
19
  pytorch-cuda==11.8 \
20
- cudatoolkit==11.8
21
 
22
  COPY ./install_deps.sh /code/install_deps.sh
23
 
 
12
  RUN conda env create -f /code/environment_docker.yml
13
 
14
  # Activate the environment and install torch, torchvision, and torchaudio
15
+ RUN /bin/bash -c "source activate masaenv && conda install -n masaenv -c pytorch -c nvidia -c conda-forge \
16
  pytorch==2.1.2 \
17
  torchvision==0.16.2 \
18
  torchaudio==2.1.2 \
19
  pytorch-cuda==11.8 \
20
+ cudatoolkit==11.8"
21
 
22
  COPY ./install_deps.sh /code/install_deps.sh
23