From 44743f03f258d5b836e9b08d61337d7855a568f8 Mon Sep 17 00:00:00 2001 From: ct00659um <ct00659@surrey.ac.uk> Date: Mon, 6 Nov 2023 17:35:42 +0000 Subject: [PATCH] Working dockerfile - need to find correct cuda container still --- Dockerfile | 16 +++++++++------- 1 file changed, 9 insertions(+), 7 deletions(-) diff --git a/Dockerfile b/Dockerfile index 5929097..3168cb1 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 -- GitLab