Update docker file with nvidia container

.dockerignore

+.venv
+dataset
\ No newline at end of file

Dockerfile

-FROM python:3.9-slim
+FROM nvcr.io/nvidia/pytorch:21.05-py3
 WORKDIR /app
+# Install opencv dependencies
+RUN apt-get update && apt-get install -y ffmpeg libsm6 libxext6
 COPY requirements.txt requirements.txt
-RUN pip install -r requirements.txt
+RUN pip install --no-cache-dir -r requirements.txt
+COPY . .
 CMD python cikm_inter_dst_predrnn_run_taasss.py
\ No newline at end of file