Update Dockerfile

Dockerfile

@@ -10,13 +10,16 @@ RUN apt-get update && apt-get install --no-install-recommends -y \
   python3-pip \
   git \
   ffmpeg \
-  && apt-get clean && rm -rf /var/lib/apt/lists/*
-  export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:`python3 -c 'import os; import torch; print(os.path.dirname(torch.__file__) +"/lib")'`
+  libcudnn8 \
+  && apt-get clean && rm -rf /var/lib/apt/lists/* 
 
 WORKDIR /code
 
 COPY ./requirements.txt /code/requirements.txt
 
+# make sure cudnn errors go away
+RUN export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:`python3 -c 'import os; import torch; print(os.path.dirname(torch.__file__) +"/lib")'`
+
 # Set up a new user named "user" with user ID 1000
 RUN useradd -m -u 1000 user
 # Switch to the "user" user