Remove pyvirtualdisplay from environment.yml and add it to the Dockerfile

docker/Dockerfile

@@ -12,6 +12,7 @@ RUN apt-get update && apt-get upgrade -y \
 
 COPY environment.yml /tmp/
 RUN conda update -y -n base conda \
+    && echo '    - pyvirtualdisplay' >> /tmp/environment.yml \
     && conda env create -f /tmp/environment.yml \
     && conda clean -y -t \
     && rm /tmp/environment.yml

environment.yml

@@ -36,7 +36,6 @@ dependencies:
     - opencv-python==4.1.1.26
     - psutil==5.6.3
     - pyglet==1.3.2
-    - pyvirtualdisplay==0.2.4
     - tensorboard==2.0.0
     - tensorflow==2.0.0
     - tensorflow-addons==0.6.0