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