From 277ed30d0c3010f70d0faff48b8d9d7b6c636f34 Mon Sep 17 00:00:00 2001
From: "Hieber, Felix" <fhieber@amazon.de>
Date: Fri, 3 Nov 2017 16:27:46 +0100
Subject: [PATCH] Remove libgomp1 from sockeye deps docker images. Change back
 to mxnet-cu90 without MKL due to
 https://github.com/apache/incubator-mxnet/issues/8532

cr https://cr.amazon.com/r/7923806/
---
 docker/Dockerfile.deps.stable | 3 +--
 docker/Dockerfile.sockeye     | 2 +-
 docker/build-deps.sh          | 2 +-
 3 files changed, 3 insertions(+), 4 deletions(-)

diff --git a/docker/Dockerfile.deps.stable b/docker/Dockerfile.deps.stable
index 648116600..97ec78869 100644
--- a/docker/Dockerfile.deps.stable
+++ b/docker/Dockerfile.deps.stable
@@ -11,7 +11,6 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
     curl \
     git \
     less \
-    libgomp1 \
     nfs-common \
     unzip \
     vim \
@@ -60,7 +59,7 @@ ENV LD_LIBRARY_PATH ${LD_LIBRARY_PATH}:/usr/local/lib:/usr/local/cuda/lib64:/usr
 RUN pip install \
     boto3 \
     docker \
-    mxnet-cu90mkl==0.12.0 \
+    mxnet-cu90==0.12.0 \
     tensorboard \
     pyaml
 
diff --git a/docker/Dockerfile.sockeye b/docker/Dockerfile.sockeye
index e86da450c..99c6910dd 100644
--- a/docker/Dockerfile.sockeye
+++ b/docker/Dockerfile.sockeye
@@ -3,7 +3,7 @@
 # If you need the latest MXNet (e.g. for experimentation), change the following to use the nightly built image
 # that includes latest MXNet
 
-FROM 405626328856.dkr.ecr.us-east-1.amazonaws.com/sockeye-deps:cu90mkl-0.12.0
+FROM 405626328856.dkr.ecr.us-east-1.amazonaws.com/sockeye-deps:cu90-0.12.0
 
 RUN sudo mkdir -p /efs && \
     sudo chown -R ec2-user:ec2-user /efs
diff --git a/docker/build-deps.sh b/docker/build-deps.sh
index 96d87afe7..6f417aaee 100755
--- a/docker/build-deps.sh
+++ b/docker/build-deps.sh
@@ -5,7 +5,7 @@
 
 ECR="405626328856.dkr.ecr.us-east-1.amazonaws.com"
 REPOSITORY="sockeye-deps"
-TAG="cu90mkl-0.12.0"
+TAG="cu90-0.12.0"
 
 DOCKERFILE=$(pwd)"/docker/Dockerfile.deps.stable"