JohanDL commited on
Commit
c994125
·
verified ·
1 Parent(s): 39a5a71

update docker

Browse files
Files changed (1) hide show
  1. Dockerfile +2 -1
Dockerfile CHANGED
@@ -10,13 +10,14 @@ COPY ./environment_docker.yml /code/environment_docker.yml
10
 
11
  # Create the environment using the environment.yml file
12
  RUN conda env create -f /code/environment_docker.yml
13
-
14
  COPY ./install_deps.sh /code/install_deps.sh
15
 
16
  # Make the install_deps.sh script executable
17
  RUN chmod +x /code/install_deps.sh
18
 
19
  RUN pip install -U openmim
 
20
  RUN mim install mmengine
21
  RUN pip install mmcv==2.1.0 -f https://download.openmmlab.com/mmcv/dist/cu118/torch2.1/index.html
22
  RUN pip install git+https://github.com/open-mmlab/mmdetection.git@v3.3.0
 
10
 
11
  # Create the environment using the environment.yml file
12
  RUN conda env create -f /code/environment_docker.yml
13
+ RUN conda activate masaenv
14
  COPY ./install_deps.sh /code/install_deps.sh
15
 
16
  # Make the install_deps.sh script executable
17
  RUN chmod +x /code/install_deps.sh
18
 
19
  RUN pip install -U openmim
20
+ RUN pip install torch torchvision torchaudio
21
  RUN mim install mmengine
22
  RUN pip install mmcv==2.1.0 -f https://download.openmmlab.com/mmcv/dist/cu118/torch2.1/index.html
23
  RUN pip install git+https://github.com/open-mmlab/mmdetection.git@v3.3.0