diff --git a/Dockerfile b/Dockerfile index c431d35a81313e813b6f79133c42fbcfd681ae2d..d16f1d40e58482e49ba267c6c85748a2bbc02606 100644 --- a/Dockerfile +++ b/Dockerfile @@ -46,6 +46,6 @@ RUN conda update -c defaults conda \ && bash RUN conda install -c conda-forge python=3.9 -RUN conda env update --file MSTpp_cuda.yml --prune && conda clean -y --all +RUN conda env update --file segmentAnything.yml --prune && conda clean -y --all RUN conda install pytorch torchvision torchaudio pytorch-cuda=11.7 -c pytorch -c nvidia