Update Dockerfile

Dockerfile

@@ -14,11 +14,7 @@ RUN apt-get update && DEBIAN_FRONTEND=noninteractive apt-get install -y --no-ins
     libxext6 \
     libgomp1 \
     libfontconfig1 \
-    libegl1-mesa-dev \
-    libgl1-mesa-dev \
     libgl1-mesa-glx \
-    libglapi-mesa \
-    libgles2-mesa-dev \
     libglib2.0-dev \
     libxrender1 \
     libgbm1 \
@@ -27,7 +23,6 @@ RUN apt-get update && DEBIAN_FRONTEND=noninteractive apt-get install -y --no-ins
     python3.8 \
     python3-pip \
     python-is-python3 \
-    python3-opencv \
     nvidia-cuda-toolkit \
     && rm -rf /var/lib/apt/lists/*
 
@@ -50,7 +45,6 @@ ENV PATH=${CUDA_HOME}/bin:/home/${USER_NAME}/.local/bin:/usr/bin:${PATH}
 ENV LD_LIBRARY_PATH=${CUDA_HOME}/lib64:/usr/local/lib/python3.8/site-packages/open3d:${LD_LIBRARY_PATH}
 ENV LIBRARY_PATH=${CUDA_HOME}/lib64/stubs:/usr/local/lib/python3.8/site-packages/open3d:${LIBRARY_PATH}
 
-
 # Set home to the user's home directory
 ENV HOME=/home/user \
     PATH=/home/user/.local/bin:$PATH \
@@ -75,6 +69,11 @@ RUN chmod 777 $HOME
 ENV TRANSFORMERS_CACHE=/tmp
 ENV MPLCONFIGDIR=/tmp
 
+# cannot cache function '_make_tree': no locator available for file '/usr/local/lib/python3.8/site-packages/pymatting/util/kdtree.py'
+ENV NUMBA_CACHE_DIR=/tmp/numba_cache
+ENV HF_HOME=$HOME/.cache/huggingface
+RUN mkdir -p $HF_HOME
+
 # Copy the current directory contents into the container at $HOME/app setting the owner to the user
 COPY --chown=user . $HOME/app