Skip to content

Commit

Permalink
[Enhancement] Use openmim in dockerfile (#107)
Browse files Browse the repository at this point in the history
  • Loading branch information
KKIEEK authored Dec 19, 2022
1 parent efa59d6 commit 58bf811
Showing 1 changed file with 4 additions and 4 deletions.
8 changes: 4 additions & 4 deletions docker/Dockerfile
Original file line number Diff line number Diff line change
Expand Up @@ -17,11 +17,11 @@ RUN apt-get update && apt-get install -y ffmpeg libsm6 libxext6 git ninja-build
&& apt-get clean \
&& rm -rf /var/lib/apt/lists/*

# Install MMCV
ARG PYTORCH
ARG CUDA
# Install MMCV with MIM
ARG MMCV_VERSION
RUN ["/bin/bash", "-c", "pip install --no-cache-dir mmcv-full==${MMCV_VERSION} -f https://download.openmmlab.com/mmcv/dist/cu${CUDA//./}/torch${PYTORCH}/index.html"]
RUN pip install --no-cache-dir --upgrade pip \
&& pip install --no-cache-dir openmim \
&& mim install mmcv-full==${MMCV_VERSION}

# Install SIATune
COPY . /siatune
Expand Down

0 comments on commit 58bf811

Please sign in to comment.