first Dockerfile that builds an image that starts training

Dockerfile

@@ -1,33 +1,33 @@
-FROM nvidia/cuda:10.0-cudnn7-devel-ubuntu16.04
+FROM nvidia/cudagl:9.0-devel-ubuntu16.04
 
 ARG DEBIAN_FRONTEND=noninteractive
 
 # Essentials: developer tools, build tools, OpenBLAS
 RUN apt-get update && apt-get install -y --no-install-recommends \
     apt-utils git curl vim unzip openssh-client wget \
     build-essential cmake \
-    libopenblas-dev
+    libopenblas-dev \
+    libglib2.0-0 \
+    libsm6 \
+    libxext6 \
+    libxrender-dev
 
 # Python 3.5
-# For convenience, alias (but don't sym-link) python & pip to python3 & pip3 as recommended in:
-# http://askubuntu.com/questions/351318/changing-symlink-python-to-python3-causes-problems
 RUN apt-get update && apt-get install -y --no-install-recommends python3.5 python3.5-dev python3-pip python3-tk && \
     pip3 install --no-cache-dir --upgrade pip setuptools && \
     echo "alias python='python3'" >> /root/.bash_aliases && \
     echo "alias pip='pip3'" >> /root/.bash_aliases
-# Pillow and it's dependencies
-RUN apt-get update && apt-get install -y --no-install-recommends libjpeg-dev zlib1g-dev && \
-    pip3 --no-cache-dir install Pillow
+
 # Science libraries and other common packages
 RUN pip3 --no-cache-dir install \
-    numpy scipy pyyaml cffi pyyaml matplotlib Cython requests
+    numpy scipy pyyaml cffi pyyaml matplotlib Cython requests opencv-python "pillow<7"
 
-# Tensorflow 1.11 - GPU
-RUN pip3 install https://download.pytorch.org/whl/cu100/torch-1.0.1.post2-cp35-cp35m-linux_x86_64.whl
-RUN pip3 install torchvision==0.2.2.post3
+# Tensorflow
+RUN pip3 install https://download.pytorch.org/whl/cu90/torch-0.4.1-cp35-cp35m-linux_x86_64.whl && \
+    pip3 install torchvision==0.2.2.post3
 
 # Expose port for TensorBoard
 EXPOSE 6006
 
 # cd to home on login
-RUN echo "cd /root/dense_fusion" >> /root/.bashrc 
+RUN echo "cd /root/dense_fusion" >> /root/.bashrc