From 23c88fe1e0a6d5b76ec702748932ccf8a9f1dabe Mon Sep 17 00:00:00 2001
From: Vignesh B <vigneshb.hp@gmail.com>
Date: Thu, 2 May 2019 19:41:46 -0400
Subject: [PATCH 1/3] Mount host source parent directory in rocm-user home

---
 docker-compose.yml | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/docker-compose.yml b/docker-compose.yml
index 23b10a9..d7e4c1a 100644
--- a/docker-compose.yml
+++ b/docker-compose.yml
@@ -69,6 +69,7 @@ services:
       - "/dev/kfd"
     image: rocm/rocm-terminal
     volumes:
+      - ..:/home/rocm-user
       - ~:/usr/local/src/host-home
 
   rocm-from-src:
@@ -79,6 +80,7 @@ services:
       - "/dev/kfd"
     image: rocm/rocm-terminal
     volumes:
+      - ..:/home/rocm-user
       - ~:/usr/local/src/host-home
     volumes_from:
       - roct-src:ro

From 321b747c1cb189376744c91696689eda6b185e9f Mon Sep 17 00:00:00 2001
From: Vignesh B <vigneshb.hp@gmail.com>
Date: Thu, 2 May 2019 20:42:56 -0400
Subject: [PATCH 2/3] Include Python3 and OpenCV git

---
 docker-compose.yml       |  4 ++++
 rocm-terminal/Dockerfile | 39 +++++++++++++++++++++++++++++++++++++++
 2 files changed, 43 insertions(+)

diff --git a/docker-compose.yml b/docker-compose.yml
index d7e4c1a..39983ea 100644
--- a/docker-compose.yml
+++ b/docker-compose.yml
@@ -71,6 +71,8 @@ services:
     volumes:
       - ..:/home/rocm-user
       - ~:/usr/local/src/host-home
+    environment:
+      - DISPLAY=${DISPLAY}
 
   rocm-from-src:
     build:
@@ -87,6 +89,8 @@ services:
       - rocr-src:ro
       - hcc-lc-src:ro
       # - hcc-hsail-src:ro
+    environment:
+      - DISPLAY:${DISPLAY}
 
   dev-ubuntu:
     build:
diff --git a/rocm-terminal/Dockerfile b/rocm-terminal/Dockerfile
index f11dc63..4b6e4e8 100644
--- a/rocm-terminal/Dockerfile
+++ b/rocm-terminal/Dockerfile
@@ -40,6 +40,45 @@ COPY sudo-nopasswd /etc/sudoers.d/sudo-nopasswd
 RUN useradd --create-home -G sudo --shell /bin/bash rocm-user
 #    sed --in-place=.rocm-backup 's|^\(PATH=.*\)"$|\1:/opt/rocm/bin"|' /etc/environment
 
+# Install Python 3
+RUN apt-get update \
+  && apt-get install -y python3-pip python3-dev python3-numpy\
+  && pip3 install --upgrade pip
+
+# Install OpenCV dependencies
+RUN apt-get install -y \
+	build-essential \
+	cmake \
+	qt5-default \
+	libvtk6-dev \
+	zlib1g-dev \
+	libjpeg-dev \
+	libwebp-dev \
+	libpng-dev \
+	libswscale-dev \
+	libtheora-dev \
+	libvorbis-dev \
+	libxvidcore-dev \
+	libx264-dev \
+	yasm \
+	libopencore-amrnb-dev \
+	libv4l-dev \
+	libxine2-dev \
+	libtbb-dev \
+	libeigen3-dev \
+	doxygen
+
+# Install OpenCV
+RUN git clone --depth 1 https://github.com/opencv/opencv.git /root/opencv && \
+	cd /root/opencv && \
+	mkdir build && \
+	cd build && \
+	cmake -DWITH_QT=ON -DWITH_OPENGL=ON -DFORCE_VTK=ON -DWITH_TBB=ON -DWITH_GDAL=ON -DWITH_XINE=ON -DBUILD_EXAMPLES=ON .. && \
+	make -j"$(nproc)"  && \
+	make install && \
+	ldconfig && \
+	echo 'ln /dev/null /dev/raw1394' >> ~/.bashrc
+
 USER rocm-user
 WORKDIR /home/rocm-user
 ENV PATH "${PATH}:/opt/rocm/bin"

From 200e916713525661aa4d30bea7e24903e2fc6b32 Mon Sep 17 00:00:00 2001
From: Vignesh Balasubramaniam <vbalasubrama@Lenovo.com>
Date: Tue, 14 May 2019 13:26:39 -0400
Subject: [PATCH 3/3] Include tensorflow-rocm

---
 rocm-terminal/Dockerfile | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/rocm-terminal/Dockerfile b/rocm-terminal/Dockerfile
index 4b6e4e8..fba67af 100644
--- a/rocm-terminal/Dockerfile
+++ b/rocm-terminal/Dockerfile
@@ -79,6 +79,9 @@ RUN git clone --depth 1 https://github.com/opencv/opencv.git /root/opencv && \
 	ldconfig && \
 	echo 'ln /dev/null /dev/raw1394' >> ~/.bashrc
 
+# Install ROCm Tensorflow
+RUN apt-get install -y rocm-libs miopen-hip cxlactivitylogger && pip3 install tensorflow-rocm
+
 USER rocm-user
 WORKDIR /home/rocm-user
 ENV PATH "${PATH}:/opt/rocm/bin"