Update gpu.Dockerfile

gpu.Dockerfile

-FROM nvcr.io/nvidia/cuda:11.2.0-cudnn8-devel-ubuntu20.04
+FROM psoto87/pytorch1.13
 
 ENV DEBIAN_FRONTEND=noninteractive
 
@@ -24,6 +24,8 @@ RUN apt-get -y update && apt-get install -y \
     python3-pip \
     python3-setuptools
 
+RUN pip install einops oss2
+
 # some image/media dependencies
 RUN apt-get -y update && apt-get install -y \
     libjpeg8-dev \
@@ -36,13 +38,13 @@ RUN apt-get -y update && apt-get install -y \
     libdc1394-22-dev \
     libxine2-dev \
     libavfilter-dev  \
-    libavutil-dev
+    libavutil-dev \
+    libavdevice-dev
 
 RUN apt-get -y update && apt-get install -y ffmpeg  
 
 RUN apt-get clean && rm -rf /tmp/* /var/tmp/* /var/lib/apt/lists/* && apt-get -y autoremove
 
-ENV NVIDIA_DRIVER_CAPABILITIES=all
-RUN ln -s /usr/lib/x86_64-linux-gnu/libnvcuvid.so.1 /usr/local/cuda/lib64/libnvcuvid.so
 RUN git clone --recursive https://github.com/dmlc/decord
-RUN cd decord && mkdir build && cd build && cmake .. -DUSE_CUDA=ON -DCMAKE_BUILD_TYPE=Release && make -j2 && cd ../python && python3 setup.py install
+RUN cd decord && mkdir build && cd build && cmake .. -DUSE_CUDA=0 -DCMAKE_BUILD_TYPE=Release && make -j2 
+RUN cd /workspace/decord/python && python3 setup.py install --user