diff --git a/.gitlab-ci.yml b/.gitlab-ci.yml new file mode 100644 index 0000000000000000000000000000000000000000..fa0c0b6847544f2f04f68d04cbde3e16b53bf03d --- /dev/null +++ b/.gitlab-ci.yml @@ -0,0 +1,4 @@ +include: + - project: 'surrey-shared-containers/templates/docker-build-ci' + file: 'docker-build-ci.yml' + \ No newline at end of file diff --git a/Dockerfile b/Dockerfile new file mode 100644 index 0000000000000000000000000000000000000000..a4327e13738f36d81f4b98c0d00800c9b246310b --- /dev/null +++ b/Dockerfile @@ -0,0 +1,55 @@ +FROM nvidia/cuda:11.2.2-cudnn8-devel-ubuntu20.04 + +ARG DEBIAN_FRONTEND=noninteractive + +RUN apt-get -y update + +RUN apt-get install -y \ + build-essential \ + cmake \ + ninja-build \ + git \ + wget \ + rsync \ + ffmpeg \ + htop \ + nano \ + libatlas-base-dev \ + libboost-all-dev \ + libeigen3-dev \ + libhdf5-serial-dev \ + libleveldb-dev \ + liblmdb-dev \ + libopenblas-dev \ + libopenblas-base \ + libsm6 \ + libxext6 \ + libxrender-dev \ + glibc-source + +RUN apt-get autoremove -y && \ + apt-get autoclean -y && \ + apt-get clean -y && \ + rm -rf /var/lib/apt/lists/* + +ENV WRKSPCE="/workspace" + +RUN wget -q https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh \ + && bash Miniconda3-latest-Linux-x86_64.sh -b -p $WRKSPCE/miniconda3 \ + && rm -f Miniconda3-latest-Linux-x86_64.sh + +ENV PATH="$WRKSPCE/miniconda3/bin:${PATH}" + +RUN conda update -c defaults conda + +RUN conda create -n segment_anything_env python=3.9 + +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