From 300c94165df1c13594fa769053997221c4a239cb Mon Sep 17 00:00:00 2001
From: Huck Febbo <hfebbo@sv802655lx.am.mds.honda.com>
Date: Fri, 31 Jan 2020 11:52:38 -0800
Subject: [PATCH] first Dockerfile that builds an image that starts training
---
Dockerfile | 24 ++++++++++++------------
1 file changed, 12 insertions(+), 12 deletions(-)
diff --git a/Dockerfile b/Dockerfile
index c1b3c77..4c7ed8b 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -1,4 +1,4 @@
-FROM nvidia/cuda:10.0-cudnn7-devel-ubuntu16.04
+FROM nvidia/cudagl:9.0-devel-ubuntu16.04
ARG DEBIAN_FRONTEND=noninteractive
@@ -6,28 +6,28 @@ ARG DEBIAN_FRONTEND=noninteractive
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
--
GitLab