diff --git a/Dockerfile b/Dockerfile index 259f98960085d803b3c1bb815a0a0fb7fd2d60c3..b2cbda392f8404fe04d0383218177a6fe435c904 100644 --- a/Dockerfile +++ b/Dockerfile @@ -27,7 +27,8 @@ RUN pip install mmcv==2.1.0 -f https://download.openmmlab.com/mmcv/dist/cu121/to RUN python -c 'import mmcv;print(mmcv.__version__)' # Install FFCV -RUN conda install cupy pkg-config libjpeg-turbo opencv cudatoolkit=${CUDA} numba -c pytorch -c conda-forge +RUN conda update -y conda && conda install -y cupy compilers pkg-config libjpeg-turbo opencv cudatoolkit=${CUDA} numba -c pytorch -c conda-forge\ + && conda clean -a RUN pip install ffcv \ && pip cache purge