update dockerfile

Author: jankinf

.dockerignore

@@ -0,0 +1 @@
+.venv/

.gitignore

@@ -186,6 +186,7 @@ comp/
 logs/
 tmpdata/
 slurm_run/
+env/apt/
 data
 playground
 log_reload.py

README.md

@@ -85,22 +85,21 @@ The envionment of this version has been updated to accommodate more latest model
     - Commands to install environment
     ```shell
     #  Note: 
-    # [code] is an `absolute path` of project root: abspath(./)
-    # [data] and [playground] are `absolute paths` of data and model_playground(decompress our provided data/playground).
+    # [data] is the `absolute paths` of data.
     
-    docker build -t multitrust:v0.0.1 -f env/Dockerfile .
+    cp -r /etc/apt ./env/apt
+
+    docker build --network=host -t multitrust:v0.0.1 -f env/Dockerfile .
 
     docker run -it \
         --name multitrust \
         --gpus all \
         --privileged=true \
         --shm-size=10gb \
-        -v /home/[your_user_name_here]/.cache/huggingface:/root/.cache/huggingface \
-        -v /home/[your_user_name_here]/.cache/torch:/root/.cache/torch \
-        -v [code]:/root/multitrust \
-        -v [data]:/root/multitrust/data \
-        -v [playground]:/root/multitrust/playground \
-        -w /root/multitrust \
+        -v $HOME/.cache/huggingface:/root/.cache/huggingface \
+        -v $HOME/.cache/torch:/root/.cache/torch \
+        -v [data]:/root/MMTrustEval/data \
+        -w /root/MMTrustEval \
         -p 11180:22 \
         -p 8000:8000 \
         -d multitrust:v0.0.1 /bin/bash

env/Dockerfile

@@ -1,41 +1,64 @@
-FROM ubuntu:22.04
+FROM nvidia/cuda:12.4.0-devel-ubuntu22.04
+
+# Set environment variable
 ENV DEBIAN_FRONTEND=noninteractive
 ENV CRYPTOGRAPHY_DONT_BUILD_RUST=1
 ENV LD_LIBRARY_PATH=/usr/local/cuda/lib64
-ENV PATH=/usr/local/cuda/bin:${PATH}
-ENV PATH=/root/miniconda3/bin:${PATH}
-# ENV http_proxy=http://your_proxy_here \
-#     https_proxy=http://your_proxy_here
+ENV PATH=/usr/local/cuda/bin:/root/miniconda3/bin:${PATH}
 
-WORKDIR /root/
+# Copy the configuration file
+COPY env/apt /etc/apt
+COPY . /root/MMTrustEval
 
-COPY ./env/sources.list /etc/apt/sources.list
-COPY ./env/requirements.txt /root/requirements.txt
+WORKDIR /root/MMTrustEval
 
 SHELL ["/bin/bash", "--login", "-c"]
 
+# System package installation and basic configuration
 RUN apt-get clean all \
+    && rm -rf /var/lib/apt/lists/* \
     && apt-get update -y \
-    && apt-get install -y apt-utils rsync openssh-server inetutils-ping wget curl sudo git zip unzip vim \
-    && apt-get install -y libgl1 gcc g++ build-essential pkg-config libssl-dev \
-    && /etc/init.d/ssh start \
+    && apt-get install -y --no-install-recommends \
+        apt-utils \
+        rsync \
+        openssh-server \
+        inetutils-ping \
+        wget \
+        curl \
+        sudo \
+        git \
+        zip \
+        unzip \
+        vim \
+        libgl1 \
+        gcc \
+        g++ \
+        build-essential \
+        pkg-config \
+        libssl-dev \
+        ca-certificates \
+        python3.9 \
+    && rm -rf /var/lib/apt/lists/*
+
+# SSH configuration
+RUN /etc/init.d/ssh start \
     && echo "PermitRootLogin yes" >> /etc/ssh/sshd_config \
     && echo 'root:123456' | chpasswd \
-    && service ssh restart \
-    && mkdir -p ~/miniconda3 \
-    && wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh -O /root/miniconda3/miniconda.sh \
-    && bash ~/miniconda3/miniconda.sh -b -u -p ~/miniconda3 \
-    && rm -rf ~/miniconda3/miniconda.sh \
-    && /root/miniconda3/bin/conda init bash \
-    && curl https://sh.rustup.rs -sSf | sh -s -- -y \
-    && bash -l -c "source /root/.bashrc" \
-    && conda install python=3.9 -y \
-    && wget https://developer.download.nvidia.com/compute/cuda/12.4.0/local_installers/cuda_12.4.0_550.54.14_linux.run -O /root/cuda.run \
-    && bash /root/cuda.run --silent --toolkit --override \
-    && rm -f /root/cuda.run \
-    && pip install -r /root/requirements.txt
-
-# for ssh login
-EXPOSE 22
-# for mkdocs service
-EXPOSE 8000 
+    && service ssh restart
+
+
+# UV
+RUN wget -qO- https://astral.sh/uv/install.sh | sh \
+    && source $HOME/.local/bin/env \
+    && uv venv --python 3.9 \
+    && source .venv/bin/activate \
+    && uv pip install setuptools \
+    && uv pip install torch==2.3.0 \
+    && uv pip sync --no-build-isolation ./env/requirements.txt
+
+
+# Port
+EXPOSE 22 8000
+
+# SSH
+CMD ["/usr/sbin/sshd", "-D"]