dockerfile with sam2

Author: xiu-cs

Dockerfile.sam2

@@ -0,0 +1,57 @@
+FROM pytorch/pytorch:2.3.1-cuda11.8-cudnn8-runtime
+ARG user_name
+ARG uid
+ARG gid
+ENV HOME=/root 
+WORKDIR ${HOME}
+
+ENV DEBIAN_FRONTEND=noninteractive
+RUN apt-get update -y && apt-get install -y --no-install-recommends \
+    vim zsh tmux wget htop jupyter python3 python3-pip libgl1-mesa-glx git sudo ssh libglib2.0-0 \
+    tree \
+    && apt-get -y autoclean \
+    && apt-get -y autoremove \
+    && rm -rf /var/lib/apt/lists/* /tmp/* /var/tmp/* \
+    && apt-get clean \
+    && apt-get update 
+
+# Install conda packages as root
+SHELL ["/bin/bash", "-lc"]
+RUN conda install -y python=3.10 pytables=3.8.0 hdf5 jupyter
+
+# Install DeepLabCut and other pip packages as root
+RUN pip install --no-cache-dir --upgrade pip setuptools && \
+    pip install --no-cache-dir gpustat nvitop pytest PySide6==6.3.1 streamlit networkx isort black && \
+    pip install --no-cache-dir amadeusgpt && \
+    pip install --no-cache-dir git+https://github.com/DeepLabCut/DeepLabCut.git
+
+# Initialize conda for zsh shell
+RUN /opt/conda/bin/conda init zsh
+
+# Create a non-root user
+RUN mkdir /app /logs /data
+RUN groupadd -g ${gid} ${user_name} \
+    && useradd -m -u ${uid} -g ${gid} ${user_name} \
+    && chown -R ${uid}:${gid} /home
+
+# Switch to the new user and set home directory as working directory
+USER ${user_name}
+ENV HOME=/home/${user_name}
+WORKDIR ${HOME}
+
+RUN git clone https://github.com/facebookresearch/sam2.git && cd sam2 && pip install -e .
+
+# Install Oh My Zsh and plugins
+# RUN sh -c "$(curl -fsSL https://raw.githubusercontent.com/ohmyzsh/ohmyzsh/master/tools/install.sh)" --unattended \
+#     && git clone https://github.com/zsh-users/zsh-autosuggestions ~/.oh-my-zsh/custom/plugins/zsh-autosuggestions \
+#     && git clone https://github.com/zsh-users/zsh-syntax-highlighting ~/.oh-my-zsh/custom/plugins/zsh-syntax-highlighting \
+#     && sed -i 's/^plugins=(.*)$/plugins=(git zsh-autosuggestions zsh-syntax-highlighting)/' ~/.zshrc
+# RUN echo "export PATH=$PATH:/home/${user_name}/.local/bin" >> /home/${user_name}/.zshrc
+
+USER root
+WORKDIR /app
+
+# CMD ["zsh"]
+# SHELL ["/bin/zsh", "-c"]
+CMD ["/bin/bash"]
+SHELL ["/bin/bash", "-c"]

Makefile

@@ -3,9 +3,9 @@ app:
 
 	streamlit run amadeusgpt/app.py --server.fileWatcherType none --server.maxUploadSize 1000
 
-IMG_TAG := 0.1
+IMG_TAG := 0.2
 IMG_NAME := amadeusgpt
-DOCKERFILE := Dockerfile
+DOCKERFILE := Dockerfile.sam2
 
 BUILD_ARGS := \
         --build-arg uid=$(shell id -u) \
@@ -24,7 +24,7 @@ DOCKER_SRC := /home/$(shell id -un)/AmadeusGPT
 VOLUMES := \
 	--volume $(HOST_SRC):$(DOCKER_SRC) 
 
-CONTAINER_TAG :=v0.1
+CONTAINER_TAG :=v0.14
 CONTAINER_NAME := amadeusgpt_$(CONTAINER_TAG)
 
 run: