Add docker image for sirius dependencies

Author: Yifei-yang7

docker/Dockerfile

@@ -1,29 +1,32 @@
-FROM nvidia/cuda:12.5.1-cudnn-devel-ubuntu22.04
+FROM nvidia/cuda:12.6.1-devel-ubuntu22.04
+ENV DEBIAN_FRONTEND=noninteractive
+WORKDIR /root
 
-ENV TZ=Asia/Dubai
-RUN ln -snf /usr/share/zoneinfo/$TZ /etc/localtime && echo $TZ > /etc/timezone
+# Common dependencies
+RUN apt-get update && apt-get install -y git g++ cmake ninja-build libssl-dev wget vim unzip
 
-RUN apt update -y
-RUN apt-get update -y
-RUN apt install -y g++ make wget unzip vim htop git dstat
-RUN apt install -y software-properties-common
-RUN add-apt-repository -y ppa:ubuntu-toolchain-r/test
-RUN apt update
-RUN apt-get install -y build-essential
-RUN apt-get install -y libc-dev
-RUN update-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-11 11
-RUN update-alternatives --install /usr/bin/g++ g++ /usr/bin/g++-11 11
+# CMake 3.30.4
+RUN wget https://github.com/Kitware/CMake/releases/download/v3.30.4/cmake-3.30.4-linux-x86_64.sh && \
+    chmod +x cmake-3.30.4-linux-x86_64.sh && \
+    ./cmake-3.30.4-linux-x86_64.sh --skip-license --prefix=/usr/local && \
+    rm cmake-3.30.4-linux-x86_64.sh
 
-# install cmake
-RUN wget https://github.com/Kitware/CMake/releases/download/v3.28.0/cmake-3.28.0-linux-x86_64.sh
-RUN mkdir /opt/cmake
-RUN sh cmake-3.28.0-linux-x86_64.sh --skip-license --prefix=/opt/cmake/
-RUN ln -s /opt/cmake/bin/cmake /usr/local/bin/cmake
-RUN apt install -y ninja-build
+# Install Miniconda
+ENV CONDA_DIR=/opt/conda
+RUN wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh && \
+    bash Miniconda3-latest-Linux-x86_64.sh -b -p $CONDA_DIR && \
+    rm Miniconda3-latest-Linux-x86_64.sh
+ENV PATH=$CONDA_DIR/bin:$PATH
 
-# Install nsight compute
-RUN apt-get install -y nsight-compute-2024.1.1
+# Create conda env with libcudf
+RUN conda update -n base -c defaults conda && \
+    conda create -n libcudf-env -y && \
+    conda run -n libcudf-env conda install -y -c rapidsai -c conda-forge -c nvidia rapidsai::libcudf
+ENV USE_CUDF=1
+ENV LIBCUDF_ENV_PREFIX=$CONDA_DIR/envs/libcudf-env
 
-RUN mkdir -p /working_dir
-WORKDIR /working_dir
-RUN git config --global --add safe.directory "*"
+# Activates libcudf env at runtime beginning
+RUN echo 'echo "[INFO] Initializing conda environment... please wait (conda activation can take a few seconds)"' >> ~/.bashrc && \
+    echo "source /opt/conda/etc/profile.d/conda.sh" >> ~/.bashrc && \
+    echo "conda activate libcudf-env" >> ~/.bashrc && \
+    echo 'echo "[INFO] Finished initializing conda environment"' >> ~/.bashrc

docs/README.md

@@ -26,7 +26,22 @@ For users who have access to AWS and want to launch AWS EC2 instances to run Sir
 
 Supported EC2 instances: G6.
 
-## Dependencies (Option 2): Install Manually
+## Dependencies (Option 2): Use Docker Image
+To use the docker image with dependencies fully installed:
+```
+sudo docker run --gpus all -it yifeiyang7/sirius_dependencies:latest bash
+```
+
+Make sure both `nvidia-driver` and `nvidia-container-toolkit` are installed, `nvidia-driver` can be installed by
+```
+sudo apt install nvidia-driver-535
+```
+and `nvidia-container-toolkit` can be installed following the [instructions](https://docs.nvidia.com/datacenter/cloud-native/container-toolkit/latest/install-guide.html). Then restart docker by
+```
+sudo systemctl restart docker
+```
+
+## Dependencies (Option 3): Install Manually
 
 ### Install duckdb dependencies
 ```