Merge pull request #365 from 8secz-johndpope/patch-1

bump docker base image

Dockerfile

@@ -1,4 +1,4 @@
-FROM nvcr.io/nvidia/cuda:10.0-cudnn7-runtime-ubuntu18.04
+FROM nvcr.io/nvidia/pytorch:21.02-py3
 
 RUN DEBIAN_FRONTEND=noninteractive apt-get -qq update \
  && DEBIAN_FRONTEND=noninteractive apt-get -qqy install python3-pip ffmpeg git less nano libsm6 libxext6 libxrender-dev \
@@ -9,6 +9,5 @@ WORKDIR /app
 
 RUN pip3 install --upgrade pip
 RUN pip3 install \
-  https://download.pytorch.org/whl/cu100/torch-1.0.0-cp36-cp36m-linux_x86_64.whl \
   git+https://github.com/1adrianb/face-alignment \
   -r requirements.txt