Skip to content
Snippets Groups Projects

Working dockerfile - need to find correct cuda container still

parent 37ce878b
No related branches found
No related tags found
No related merge requests found
...@@ -40,14 +40,16 @@ RUN wget -q https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.s ...@@ -40,14 +40,16 @@ RUN wget -q https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.s
ENV PATH="$WRKSPCE/miniconda3/bin:${PATH}" ENV PATH="$WRKSPCE/miniconda3/bin:${PATH}"
COPY segmentAnything_old.yml . RUN conda update -c defaults conda
RUN conda update -c defaults conda \
&& conda init \
&& bash
RUN conda install -c conda-forge python=3.9 RUN conda create -n segment_anything_env python=3.9
#RUN conda env update --file segmentAnything.yml --prune && conda clean -y --all
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 ffmpeg gnutls
#RUN pip3 install opencv-python pycocotools matplotlib onnxruntime onnx #RUN pip3 install opencv-python pycocotools matplotlib onnxruntime onnx
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment