Initial commit

Dockerfile

+FROM nvidia/cuda:12.3.1-devel-ubuntu20.04
+
+RUN ln -fs /usr/share/zoneinfo/Europe/Warsaw /etc/localtime
+RUN apt-get update && apt-get install -y --no-install-recommends \
+    git \
+    language-pack-en-base \
+    openssh-server \
+    openssh-client \
+    python3-pip \
+    python3-dev \
+    python3-setuptools \
+    ssh \
+    sudo \
+    vim \
+    wget \
+    less \
+    libglib2.0-0 \
+    libsm6 \
+    libxext6 \
+    libxrender-dev \
+    libsndfile1 \
+    && rm -rf /var/lib/apt/lists/* \
+    && python3 -m pip install --no-cache-dir --upgrade \
+    autopep8 \
+    doc8 \
+    docutils \
+    ipython \
+    pip \
+    pylint \
+    pytest \
+    rope \
+    setuptools \
+    wheel \
+    torch \
+    torchvision \
+    tqdm \
+    pandas
+    
+# OpenSSH configuration
+RUN mkdir /var/run/sshd \
+    && sed 's@session\s*required\s*pam_loginuid.so@session optional pam_loginuid.so@g' -i /etc/pam.d/sshd
+                                                    
+# Create non-root user
+ARG UID=1000
+ARG GID=1000
+ARG USERNAME=nibio
+ENV HOME /home/${USERNAME}
+
+RUN groupadd -g ${GID} ${USERNAME} \
+    && useradd -ms /bin/bash -u ${UID} -g ${GID} -G sudo ${USERNAME} \
+    && echo "${USERNAME}:${USERNAME}" | chpasswd \
+    && mkdir -p ${HOME}/data ${HOME}/.vscode-server ${HOME}/.ssh\
+    && chown ${USERNAME}:${USERNAME} ${HOME}/data ${HOME}/.vscode-server ${HOME}/.ssh
+
+VOLUME [ "${HOME}/data", "${HOME}/.ssh", "${HOME}/.vscode-server"]
+
+# Expose SSH
+EXPOSE 22
+
+CMD [ "/usr/sbin/sshd", "-D" ]

build.sh

+#/bin/sh
+
+docker build -t nibio/pointcept:latest --build-arg UID=`id -u` --build-arg GID=`id -g` --build-arg USERNAME=nibio .

install.sh

+#!/bin/bash
+export MAX_JOBS=4
+export MAKEFLAGS="-j4" 
+
+
+pip install ninja
+
+# Install PyTorch, torchvision, torchaudio with the specified versions and CUDA toolkit 11.3
+pip install torch torchvision torchaudio
+
+# Install other dependencies
+pip install h5py pyyaml
+pip install sharedarray tensorboard tensorboardx yapf addict einops scipy plyfile termcolor timm
+
+# Install PyTorch Geometric dependencies
+pip install torch-scatter torch-sparse torch-cluster 
+pip install torch-geometric
+
+# PTv1 & PTv2 or precise eval
+cd libs/pointops
+
+TORCH_CUDA_ARCH_LIST="8.0" pip install .
+
+# Open3D (visualization, optional)
+pip install open3d
+pip install flash-attn
\ No newline at end of file

run.sh

+#/bin/sh
+
+# make mutable-outside-world dir if does not exits
+DOCKER_LOCATION="/home/nibio/"
+MUTABLE_OUTSIZE_WORLD_DIR="/mnt/media/mutable-outside-world"
+mkdir -p $MUTABLE_OUTSIZE_WORLD_DIR
+
+docker run --gpus all \
+           --mount type=volume,src=nibio-cuda-vscode-data,dst=${DOCKER_LOCATION}/data \
+           --mount type=volume,src=nibio-cuda-vscode-server,dst=${DOCKER_LOCATION}/.vscode-server \
+           --mount type=volume,src=nibio-cuda-ssh,dst=${DOCKER_LOCATION}/.ssh \
+           --mount type=bind,src=${MUTABLE_OUTSIZE_WORLD_DIR},dst=${DOCKER_LOCATION}/mutable-outside-world \
+		   --name pointcept-test \
+		   --shm-size 32GB \
+		   --ipc private \
+		   --publish 127.0.0.1:49201:22 \
+		   -itd nibio/pointcept