Update Dockerfiles to include required graphviz packages

Based on success path from Issue #18, verified on Dockerfile.cpu run using 04_01_gan_camel_train

Signed-Off-By: Joe Handzik <[email protected]>

Author: joehandzik

Dockerfile.cpu

@@ -10,6 +10,10 @@ ARG groupid=1000
 ARG userid=1000
 ## end modify
 
+RUN apt-get update && apt-get install -y \
+    graphviz \
+    libgraphviz-dev
+
 COPY ./requirements.txt /
 RUN python3 -m pip install --upgrade pip
 RUN pip install -r /requirements.txt

Dockerfile.gpu

@@ -10,6 +10,10 @@ ARG groupid=1000
 ARG userid=1000
 ## end modify
 
+RUN apt-get update && apt-get install -y \
+    graphviz \
+    libgraphviz-dev
+
 COPY ./requirements.txt /
 RUN python3 -m pip install --upgrade pip
 RUN pip install -r /requirements.txt