backup Dockerfile '-'

Dockerfile

@@ -6,14 +6,23 @@ ENV DEBIAN_FRONTEND=noninteractive
 
 
 RUN apt-get update
-RUN apt-get install -y build-essential cmake neovim nano wget curl tar eza git python3-full python3-pip
-RUN pip install numpy psutil --break-system-packages
+RUN apt-get install -y build-essential llvm-dev libzstd-dev cmake neovim nano wget curl tar eza git python3-full python3-pip
+
+# Configure pip cache to be not need to be re-downloaded every time during the TVM/MLC build/installation
+RUN mkdir -p /etc/xdg/pip/
+RUN echo "[global]" >> /etc/xdg/pip/pip.conf
+RUN echo "cache-dir = /volume/.cache/pip" >> /etc/xdg/pip/pip.conf
+
+
+# RUN pip install numpy psutil decorator --break-system-packages
 
 RUN curl --proto '=https' --tlsv1.2 -sSf https://sh.rustup.rs | sh -s -- -y
 
 RUN echo 'export PATH="$HOME/.cargo/bin:$PATH"' >> ~/.bashrc && \
     echo "export MLC_LLM_HOME=/volume" >> ~/.bashrc && \
-    echo "export PYTHONPATH=/volume/python:$PYTHONPATH" >> ~/.bashrc
+    echo "export PYTHONPATH=/volume/python:$PYTHONPATH" >> ~/.bashrc \
+    echo "export TVM_HOME=/volume/3rdparty/tvm" >> ~/.bashrc && \
+    echo "export PYTHONPATH=/volume/3rdparty/tvm/python:$PYTHONPATH" >> ~/.bashrc
 
 RUN echo 'alias python=python3' >> ~/.bashrc && \
     echo 'alias pip3=pip' >> ~/.bashrc && \