Update Dockerfile

Dockerfile

@@ -24,6 +24,7 @@ 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/*
 
@@ -67,7 +68,7 @@ RUN cd /tmp && pip install -r requirements.txt
 RUN pip install --no-index --no-cache-dir pytorch3d -f https://dl.fbaipublicfiles.com/pytorch3d/packaging/wheels/py38_cu116_pyt1130/download.html
 RUN python -c "import torch; print(torch.version.cuda)"
 
-RUN chmod 777 $HOME
+RUN chmod -R 777 $HOME
 ENV TRANSFORMERS_CACHE=/tmp
 ENV MPLCONFIGDIR=/tmp