diff --git a/Dockerfile b/Dockerfile index 5929097d213437dae1e0d402e3ee98c5e25da25c..3168cb1fa8004265621e447be769a269fe94da27 100644 --- a/Dockerfile +++ b/Dockerfile @@ -40,14 +40,16 @@ RUN wget -q https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.s ENV PATH="$WRKSPCE/miniconda3/bin:${PATH}" -COPY segmentAnything_old.yml . -RUN conda update -c defaults conda \ - && conda init \ - && bash +RUN conda update -c defaults conda -RUN conda install -c conda-forge python=3.9 -#RUN conda env update --file segmentAnything.yml --prune && conda clean -y --all +RUN conda create -n segment_anything_env python=3.9 -RUN conda install pytorch>=1.7 torchvision>=0.8 torchaudio pytorch-cuda=11.7 -c pytorch -c nvidia +SHELL ["conda", "run", "-n", "segment_anything_env", "/bin/bash", "-c"] + +RUN git clone https://github.com/facebookresearch/segment-anything.git \ + && cd segment-anything && pip install -e . \ + && pip install opencv-python pycocotools matplotlib onnxruntime onnx + +# RUN conda install pytorch>=1.7 torchvision>=0.8 torchaudio pytorch-cuda=11.7 -c pytorch -c nvidia #RUN pip3 install ffmpeg gnutls #RUN pip3 install opencv-python pycocotools matplotlib onnxruntime onnx