From 7b37d1aacbe38302d84cbe636db5fadc83033e98 Mon Sep 17 00:00:00 2001
From: Jeremy Fix <jeremy.fix@centralesupelec.fr>
Date: Fri, 22 Nov 2019 10:58:53 +0100
Subject: [PATCH 1/7] copy 18.10 to 18.04

---
 tensorflow/ubuntu-18.04/.build.sh.swp      | Bin 0 -> 16384 bytes
 tensorflow/ubuntu-18.04/Dockerfile         |  27 ++
 tensorflow/ubuntu-18.04/build.sh           | 133 ++++++
 tensorflow/ubuntu-18.04/cuda.sh            | 463 +++++++++++++++++++++
 tensorflow/ubuntu-18.04/docker-compose.yml |  16 +
 5 files changed, 639 insertions(+)
 create mode 100644 tensorflow/ubuntu-18.04/.build.sh.swp
 create mode 100755 tensorflow/ubuntu-18.04/Dockerfile
 create mode 100755 tensorflow/ubuntu-18.04/build.sh
 create mode 100755 tensorflow/ubuntu-18.04/cuda.sh
 create mode 100755 tensorflow/ubuntu-18.04/docker-compose.yml

diff --git a/tensorflow/ubuntu-18.04/.build.sh.swp b/tensorflow/ubuntu-18.04/.build.sh.swp
new file mode 100644
index 0000000000000000000000000000000000000000..98bbf89f3f6d117c7f1fa918f99f29ea2d748449
GIT binary patch
literal 16384
zcmeI3UyS2M9mnSuN=yG-d!<MqUZ&ol?&)y6d)M9`+Xm!KyxDZKNl3E2F73%U&LnZy
zv5oC)ZhKk~e<YrG1O$EHp>PsBpcV<F0s;wm0Kp@o{F8Wr@<8DMA;c5nJN~n=v&-I9
z;sMdI^pWH7%x`AC^P4{&JH6$nYTNWmahc$FfROUbuU`A}!ne=-zCp;qx~@ORe17ND
zBM4pFX3nW`_|1<c29;{9tMpxSkNI-QoWS)5wmX%36U**1Uzr6dlitJ$C-Mr)_ox!x
zF9xGq<|85q1%v`~3QWR*eC0lJ>EcS%)fc6Q={p|WnTsSYg#tnWp@2|8C?FIN3J3*+
z0z!fRX9YrUfqWblTu3T;JvpD5JHI8lzArf+%-v5e#Yre26c7ps1%v`Z0il3UKqw#-
z5DEwdgaSf=|3L+eJ|Xv`T=(z|0KETCpZ{M+;2*&s!1us+!3$s?48RU}4|oWi0}p_I
zoF(Kn@Fmay4}src!{F!OXW+|V3O)c9!C&qr<U8OM@D=bPFu~p62lo*2eeepn2{wQR
zR=|5f0h|SYx|@*CfC11z1zZ5{1O)uz?XUxGf+1*u_X7#s3I28$A-@Jc1}}jxf=`1X
z(7-Zy7gzviz+dkq<VWC#;1KMBr@<9a0RMU$A+LgOfKP!v&;(0h0sQU`LVgQg1kZvG
zfd<$H9|V`cMerav2mbz6LVgLp1wIE{@Fchk{1lrqKLK9{CGZ4T1((6Q!9UTazXIO`
zp9NdsUclS>4lsYv+@WMY#0Xv24iwMwbk8vNj3HZ#zQ)iqAtaZY%CcdGmh0%u*)K`;
zie9g+x7GHp-cq}prNWK5kVA<+d(6Vf4MN?voITeIOOkAlZzqx;o)qg=Rx<Zf13lw;
zX49yisEz4518Z19Nl;M2?0^xJx3jap(X3arw%$;;wUSi0(cRFW(%PL`v!UZN7rm=i
z4`uJ5aHF-`-9%V&?@%@;eItrp-l?d$V8=<r8?*QlGHa>jEw!q}S<SwZk%!)dmnD0=
z=^HOA#!=&O%_BEEt!b?ReTGVfosOngTRRe!*+;3gBGGG)({RMUaw;mZFHC$!7HQcX
zdzNi*SLa_$5lZcTd>=pAb<g!fI_K+flc>?V&1QWIKEiv1Vq3k-kFK52FJNYocF`3W
z4W}-Jqq2Y+KBLx{8#kPg4qTs?jo;g|oFNTD<`ub1B>6YW<+@&Lbh>K2p7g({0GrKi
z?TyBQe@EsLM=lkzS2JTgnx<9suGZ)@+wE>?`S?9vhSEwd96>Xuk9C?|^FEN&2<D?2
zq@_p{O)fJLV}rM>dt@xz9;LX65{1yr+;u#|H^wYven4*NB}qZcWxd(z>Kk>n+QASu
zHhgnba^NoeOoI1pGBo6YJE5lGP}a9Xw4)W!eZ#kmo{biFX~05iI0qCXk2SWQk321i
z(oE6zXyzoRDze&acDj1Gx!tPOF(kz!?b38)nIk$h&GSd@%KXTj>jX#1(*|od%iCzI
zna)u&)~bzKS34fAUhU)pJ6cy)8<lpmR?)9E+glxsStlfGwzLLZva?&R=aNTxRV(e|
zfm`av7I&7pa&;~p4@2#9=_}2vje1k9=-XQ-<Tf986uVqk8`Yel$VrX|M}>yxt>>lC
z8tQso)1R)ZIU`Z*C$(*K#b&v*!aH|^u|99<9;1CL;O*Y0u0zYr37Z{yxp;ZL*0FrY
z&2CmUIypV!62l*Mx|&{ATk3kPUhCGhPU*4Y#rIt*J~nT;T5jbGM5X0ckXdf5%l3uT
zNHWE#?JDMX_*1*t>>l!gEgvCS0Q~0&e5nvSuT)qnBv#jJjVO%DCVeCx`e<N<>~ZS(
zmJ=@V-^C#d`8ipm1WV_iS-y7uk@Hf%0D5YxR7g69cMy(Thxd<G^b55MN3$+s_Guw0
zE=kU%%<2fMMRZ)#b{&R=LZ5j^yMkXm-*C|Lc+zBGN|yvw8ik=3tSJgojwZdL>5el7
zGe3)nAQ68|<pIX{qmN}3f-)&sM$-;sG&MGzPK&cRMb{rHmJ@`AZKr2Fa3{Wr34~7>
zV;6%@pM{2H2Su3LO~xU^4qVJN%-D-&mHfbCIQ?jNawkE?2acwoc)5h?q>~~yoEzoL
zk>Lz6vSGgJM)ldi$G>zRgDR%bqj4?j5B!5*3~;IAWfKsBr6)~lNAV@G5y*z;*_N3M
zka3XbGtYNX;Q;d@!Mu!7JWV9a2N1%UjJ*Ro9WlPv<14;Wsj#koSgWUPMq<e*U(W6u
zAxeJK3W-6}XFNQ%Cp(!4G<GbQ%rF1agwdK335|ZAa$Ov(D``|p84Wb#>x)|4fS3X(
z>gT)gcAi<{r&wHDzUz!JRKN>O6Sfj1$;9g;qHNfhui^<Oz+FC@3@==uk}?T=rDr(`
zuH*urJdP&MQ&J*_OiI(dt(2+Y*({?%WCL=RC_Yj>xfI83p-8;qGh1_|;<hr?D@m(-
z;_W~O)AfC3h7@;G|4Wu|QRkwRuNRAHj`4yZ5sC7v*~)~@uYwYvi45S2bxIA}G6EV*
z`ff?eFHcm=+WgTWs(|g9hOO{q(rqs>qVZJXi_*TqTXZdzVy(b$Nrdo5m*Gqgr*{<i
z4ZIyC<In%+&Jtn({``-}^5}RC&+}gcH^EoIC&4Gc{opKk9r3>aJ`bjVr~e2LhfqK$
zAQTV^2nB=!LII(GP(Uak6c7r$Nfj_hW4BM`HeFp^rAoBV5-4~8->|MHyDkHl|LvQ4
z%pY6Wr-?SD7vsGcnQu_c_FI%=uOZ_<g=b$wVt*pZq>z6($DjIl_)8RQJRwVLI`LN}
zd>^d4L7ynmmEsjbq8*_BYOdIG!LH47*k{lkygb00rR-}wJe-gB`j$^+@05ba@0Ie^
Nc?0hnPkDDp{tY;HQ8@qr

literal 0
HcmV?d00001

diff --git a/tensorflow/ubuntu-18.04/Dockerfile b/tensorflow/ubuntu-18.04/Dockerfile
new file mode 100755
index 0000000..538bb43
--- /dev/null
+++ b/tensorflow/ubuntu-18.04/Dockerfile
@@ -0,0 +1,27 @@
+FROM ubuntu:18.10
+
+RUN apt update && apt install -y \
+    build-essential \
+    curl \
+    git \
+    wget \
+    libjpeg-dev \
+    openjdk-8-jdk \
+    gcc-7 \
+    g++-7 \
+    && rm -rf /var/lib/lists/*
+
+# Install Anaconda
+WORKDIR /
+RUN wget "https://repo.continuum.io/miniconda/Miniconda3-latest-Linux-x86_64.sh" -O "miniconda.sh" && \
+    bash "miniconda.sh" -b -p "/conda" && \
+    rm miniconda.sh && \
+    echo PATH='/conda/bin:$PATH' >> /root/.bashrc && \
+    /conda/bin/conda config --add channels conda-forge && \
+    /conda/bin/conda update --yes -n base conda && \
+    /conda/bin/conda update --all --yes
+
+COPY build.sh /build.sh
+COPY cuda.sh /cuda.sh
+
+CMD bash build.sh
diff --git a/tensorflow/ubuntu-18.04/build.sh b/tensorflow/ubuntu-18.04/build.sh
new file mode 100755
index 0000000..02f926f
--- /dev/null
+++ b/tensorflow/ubuntu-18.04/build.sh
@@ -0,0 +1,133 @@
+#!/usr/bin/env bash
+set -e
+
+export PATH="/conda/bin:/usr/bin:$PATH"
+
+if [ "$USE_GPU" -eq "1" ]; then
+  export CUDA_HOME="/usr/local/cuda"
+  alias sudo=""
+  source cuda.sh
+  cuda.install $CUDA_VERSION $CUDNN_VERSION $NCCL_VERSION
+  cd /
+fi
+
+# Set correct GCC version
+GCC_VERSION="7"
+update-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-$GCC_VERSION 10
+update-alternatives --install /usr/bin/g++ g++ /usr/bin/g++-$GCC_VERSION 10
+update-alternatives --set gcc "/usr/bin/gcc-$GCC_VERSION"
+update-alternatives --set g++ "/usr/bin/g++-$GCC_VERSION"
+gcc --version
+
+# Install an appropriate Python environment
+conda config --add channels conda-forge
+conda create --yes -n tensorflow python==$PYTHON_VERSION
+source activate tensorflow
+conda install --yes numpy wheel bazel==$BAZEL_VERSION
+pip install keras-applications keras-preprocessing
+
+# Compile TensorFlow
+
+# Here you can change the TensorFlow version you want to build.
+# You can also tweak the optimizations and various parameters for the build compilation.
+# See https://www.tensorflow.org/install/install_sources for more details.
+
+cd /
+rm -fr tensorflow/
+git clone --depth 1 --branch $TF_VERSION_GIT_TAG "https://github.com/tensorflow/tensorflow.git"
+
+TF_ROOT=/tensorflow
+cd $TF_ROOT
+
+# Python path options
+export PYTHON_BIN_PATH=$(which python)
+export PYTHON_LIB_PATH="$($PYTHON_BIN_PATH -c 'import site; print(site.getsitepackages()[0])')"
+export PYTHONPATH=${TF_ROOT}/lib
+export PYTHON_ARG=${TF_ROOT}/lib
+
+# Compilation parameters
+export TF_NEED_CUDA=0
+export TF_NEED_GCP=1
+export TF_CUDA_COMPUTE_CAPABILITIES=5.2,3.5
+export TF_NEED_HDFS=1
+export TF_NEED_OPENCL=0
+export TF_NEED_JEMALLOC=1  # Need to be disabled on CentOS 6.6
+export TF_ENABLE_XLA=0
+export TF_NEED_VERBS=0
+export TF_CUDA_CLANG=0
+export TF_DOWNLOAD_CLANG=0
+export TF_NEED_MKL=0
+export TF_DOWNLOAD_MKL=0
+export TF_NEED_MPI=0
+export TF_NEED_S3=1
+export TF_NEED_KAFKA=1
+export TF_NEED_GDR=0
+export TF_NEED_OPENCL_SYCL=0
+export TF_SET_ANDROID_WORKSPACE=0
+export TF_NEED_AWS=0
+export TF_NEED_IGNITE=0
+export TF_NEED_ROCM=0
+
+# Compiler options
+export GCC_HOST_COMPILER_PATH=$(which gcc)
+
+# Here you can edit this variable to set any optimizations you want.
+export CC_OPT_FLAGS="-march=native"
+
+if [ "$USE_GPU" -eq "1" ]; then
+  # Cuda parameters
+  export CUDA_TOOLKIT_PATH=$CUDA_HOME
+  export CUDNN_INSTALL_PATH=$CUDA_HOME
+  export TF_CUDA_VERSION="$CUDA_VERSION"
+  export TF_CUDNN_VERSION="$CUDNN_VERSION"
+  export TF_NEED_CUDA=1
+  export TF_NEED_TENSORRT=0
+  export TF_NCCL_VERSION=$NCCL_VERSION
+  export NCCL_INSTALL_PATH=$CUDA_HOME
+  export NCCL_INSTALL_PATH=$CUDA_HOME
+
+  # Those two lines are important for the linking step.
+  export LD_LIBRARY_PATH="$CUDA_TOOLKIT_PATH/lib64:${LD_LIBRARY_PATH}"
+  ldconfig
+fi
+
+# Compilation
+./configure
+
+if [ "$USE_GPU" -eq "1" ]; then
+
+  bazel build --config=opt \
+              --config=cuda \
+              --linkopt="-lrt" \
+              --linkopt="-lm" \
+              --host_linkopt="-lrt" \
+              --host_linkopt="-lm" \
+              --action_env="LD_LIBRARY_PATH=${LD_LIBRARY_PATH}" \
+              //tensorflow/tools/pip_package:build_pip_package
+
+  PACKAGE_NAME=tensorflow-gpu
+  SUBFOLDER_NAME="${TF_VERSION_GIT_TAG}-py${PYTHON_VERSION}-cuda${TF_CUDA_VERSION}-cudnn${TF_CUDNN_VERSION}"
+
+else
+
+  bazel build --config=opt \
+              --linkopt="-lrt" \
+              --linkopt="-lm" \
+              --host_linkopt="-lrt" \
+              --host_linkopt="-lm" \
+              --action_env="LD_LIBRARY_PATH=${LD_LIBRARY_PATH}" \
+              //tensorflow/tools/pip_package:build_pip_package
+
+  PACKAGE_NAME=tensorflow
+  SUBFOLDER_NAME="${TF_VERSION_GIT_TAG}-py${PYTHON_VERSION}"
+fi
+
+mkdir -p "/wheels/$SUBFOLDER_NAME"
+
+bazel-bin/tensorflow/tools/pip_package/build_pip_package "/wheels/$SUBFOLDER_NAME" --project_name "$PACKAGE_NAME"
+
+# Use the following for TF <= 1.8
+# bazel-bin/tensorflow/tools/pip_package/build_pip_package "/wheels/$SUBFOLDER_NAME"
+
+# Fix wheel folder permissions
+chmod -R 777 /wheels/
diff --git a/tensorflow/ubuntu-18.04/cuda.sh b/tensorflow/ubuntu-18.04/cuda.sh
new file mode 100755
index 0000000..8086b01
--- /dev/null
+++ b/tensorflow/ubuntu-18.04/cuda.sh
@@ -0,0 +1,463 @@
+#!/usr/bin/env bash
+
+# Cuda and friends installation done right.
+# Switch default Cuda version using symbolic link: cuda.switch 9.2
+# Install Cuda: cuda.install.cuda 10.0
+# Install cuDNN to CUDA_HOME: cuda.install.cudnn 7.5
+# Install NCCL to CUDA_HOME: cuda.install.nccl 2.4
+# Install Cuda, cuDNN and NCCL: cuda.install 10.0 7.5 2.4
+
+# Author: Hadrien Mary <hadrien.mary@gmail.com>
+# License: MIT License
+# Date: 11/03/2019
+
+is_cuda_home_set() {
+  if [ -z "$CUDA_HOME" ]; then
+    echo "CUDA_HOME is not set. Please set it:"
+    echo 'export CUDA_HOME="/usr/local/cuda/"'
+    return 1
+  fi
+  return 0
+}
+
+is_cuda_home_symbolic_link() {
+  if [[ -e "${CUDA_HOME}" && -L "${CUDA_HOME}" ]]; then
+    return 0
+  elif [[ ! -d "${CUDA_HOME}" && ! -f "${CUDA_HOME}" ]]; then
+    return 0
+  else
+    echo "CUDA_HOME is not a symbolic link."
+    echo "Please make it a symbolic link."
+    return 1
+  fi
+}
+
+guess_cuda_version() {
+	if ! is_cuda_home_set; then
+	  return 1
+	fi
+
+	if ! is_cuda_home_symbolic_link; then
+	  return 1
+	fi
+
+	POSSIBLE_CUDA_VERSION=$(cat "$CUDA_HOME/version.txt" | cut -d' ' -f 3 | cut -d'.' -f 1-2)
+	echo $POSSIBLE_CUDA_VERSION
+}
+
+cuda.see() {
+  if ! is_cuda_home_set; then
+    return 1
+  fi
+
+  PARENT_BASE_DIR=$(dirname $CUDA_HOME)
+  ls -l $PARENT_BASE_DIR
+  return 0
+}
+
+cuda.switch() {
+  if ! is_cuda_home_set; then
+    return 1
+  fi
+
+  if ! is_cuda_home_symbolic_link; then
+    return 1
+  fi
+
+  if [ -z "$1" ]; then
+      echo "Please specify a Cuda version."
+      echo "Usage: cuda.switch CUDA_VERSION"
+      echo "Cuda version available: 9.0, 9.1, 9.2, 10.0, 10.1"
+      return 1
+  fi
+
+  NEW_CUDA_VERSION="$1"
+  NEW_CUDA_HOME="$CUDA_HOME-$NEW_CUDA_VERSION"
+
+  if [ ! -d $NEW_CUDA_HOME ]; then
+    echo "Cuda $NEW_CUDA_VERSION doesn't exist at $NEW_CUDA_HOME."
+    return 1
+  fi
+
+  PARENT_BASE_DIR=$(dirname $CUDA_HOME)
+  if [ ! -w "$PARENT_BASE_DIR" ]; then
+    sudo rm -f $CUDA_HOME
+    sudo ln -s $NEW_CUDA_HOME $CUDA_HOME
+  else
+    rm -f $CUDA_HOME
+    ln -s $NEW_CUDA_HOME $CUDA_HOME
+  fi
+  echo "Default Cuda version is now $NEW_CUDA_VERSION at $NEW_CUDA_HOME"
+}
+
+cuda.install() {
+  cuda.install.cuda $1
+  cuda.install.cudnn $2
+  cuda.install.nccl $3
+}
+
+cuda.install.cuda() {
+
+  CUDA_VERSION="$1"
+  if [ -z "$CUDA_VERSION" ]; then
+    echo "Please specify a Cuda version."
+    echo "Usage: cuda.install.cuda CUDA_VERSION"
+    echo "Example: cuda.install.cuda 10.0"
+    echo "Cuda version available: 9.0, 9.1, 9.2, 10.0, 9.2."
+    return 1
+  fi
+
+  if ! is_cuda_home_set; then
+    return 1
+  fi
+
+  if ! is_cuda_home_symbolic_link; then
+    return 1
+  fi
+
+  CUDA_PATH="$CUDA_HOME-$CUDA_VERSION"
+  if [ -d $CUDA_PATH ]; then
+    echo "$CUDA_PATH exists. Please remove the previous Cuda folder first."
+    return 1
+  fi
+
+  # Setup Cuda URL
+  if [ "$CUDA_VERSION" = "9.0" ]; then
+    CUDA_URL="https://developer.nvidia.com/compute/cuda/9.0/Prod/local_installers/cuda_9.0.176_384.81_linux-run"
+  elif [ "$CUDA_VERSION" = "9.1" ]; then
+    CUDA_URL="https://developer.nvidia.com/compute/cuda/9.1/Prod/local_installers/cuda_9.1.85_387.26_linux"
+  elif [ "$CUDA_VERSION" = "9.2" ]; then
+    CUDA_URL="https://developer.nvidia.com/compute/cuda/9.2/Prod/local_installers/cuda_9.2.88_396.26_linux"
+  elif [ "$CUDA_VERSION" = "10.0" ]; then
+    CUDA_URL="https://developer.nvidia.com/compute/cuda/10.0/Prod/local_installers/cuda_10.0.130_410.48_linux"
+  elif [ "$CUDA_VERSION" = "10.1" ]; then
+    CUDA_URL="https://developer.nvidia.com/compute/cuda/10.1/Prod/local_installers/cuda_10.1.105_418.39_linux.run"
+  else
+    echo "Error: You need to set CUDA_VERSION to 9.0, 9.1, 9.2, 10.0 or 10.1."
+    return 1
+  fi
+
+  CUDA_INSTALLER_PATH="/tmp/cuda.run"
+
+  echo "Download Cuda $CUDA_VERSION."
+  wget "$CUDA_URL" -O "$CUDA_INSTALLER_PATH"
+
+  echo "Install Cuda $CUDA_VERSION."
+  PARENT_BASE_DIR=$(dirname $CUDA_HOME)
+  if [ ! -w "$PARENT_BASE_DIR" ]; then
+    sudo bash "$CUDA_INSTALLER_PATH" --silent --toolkit --override --toolkitpath="$CUDA_PATH"
+  else
+    bash "$CUDA_INSTALLER_PATH" --silent --toolkit --override --toolkitpath="$CUDA_PATH"
+  fi
+  rm -f "$CUDA_INSTALLER_PATH"
+
+  # Set the symbolic link.
+  cuda.switch $CUDA_VERSION
+
+  echo "Cuda $CUDA_VERSION is installed at $CUDA_PATH."
+
+  return 0
+}
+
+cuda.install.cudnn() {
+  # Install cuDNN in $CUDA_HOME
+
+  if ! is_cuda_home_set; then
+    return 1
+  fi
+
+  if ! is_cuda_home_symbolic_link; then
+    return 1
+  fi
+
+  CUDA_VERSION="$(guess_cuda_version)"
+  if [ -z "$CUDA_VERSION" ]; then
+    echo "Can't guess the Cuda version from $CUDA_HOME."
+    return 1
+  fi
+
+  CUDNN_VERSION="$1"
+  if [ -z "$CUDNN_VERSION" ]; then
+    echo "Please specify a cuDNN version."
+    echo "Usage: cuda.install.cudnn CUDNN_VERSION"
+    echo "Example: cuda.install.cudnn 7.5"
+    echo "cuDNN version available: 7.0, 7.1, 7.4, 7.5."
+    return 1
+  fi
+
+  # cuDNN 7.0
+  if [ "$CUDNN_VERSION" = "7.0" ]; then
+
+    if [ "$CUDA_VERSION" = "9.0" ]; then
+      CUDNN_VERSION_DETAILED="7.0.5.15"
+    elif [ "$CUDA_VERSION" = "9.1" ]; then
+      CUDNN_VERSION_DETAILED="7.0.5.15"
+    elif [ -n "$CUDNN_VERSION" ]; then
+      echo "Error: cuDNN $CUDNN_VERSION is not compatible with Cuda $CUDA_VERSION."
+      return 1
+    fi
+
+  # cuDNN 7.1
+  elif [ "$CUDNN_VERSION" = "7.1" ]; then
+
+    if [ "$CUDA_VERSION" = "9.0" ]; then
+      CUDNN_VERSION_DETAILED="7.1.4.18"
+    elif [ "$CUDA_VERSION" = "9.2" ]; then
+      CUDNN_VERSION_DETAILED="7.1.4.18"
+    elif [ -n "$CUDNN_VERSION" ]; then
+      echo "Error: cuDNN $CUDNN_VERSION is not compatible with Cuda $CUDA_VERSION."
+      return 1
+    fi
+
+  # cuDNN 7.4
+  elif [ "$CUDNN_VERSION" = "7.4" ]; then
+
+    if [ "$CUDA_VERSION" = "9.0" ]; then
+      CUDNN_VERSION_DETAILED="7.4.2.24"
+    elif [ "$CUDA_VERSION" = "9.2" ]; then
+      CUDNN_VERSION_DETAILED="7.4.2.24"
+    elif [ "$CUDA_VERSION" = "10.0" ]; then
+      CUDNN_VERSION_DETAILED="7.4.2.24"
+    elif [ -n "$CUDNN_VERSION" ]; then
+      echo "Error: cuDNN $CUDNN_VERSION is not compatible with Cuda $CUDA_VERSION."
+      return 1
+    fi
+
+  # cuDNN 7.5
+  elif [ "$CUDNN_VERSION" = "7.5" ]; then
+
+    if [ "$CUDA_VERSION" = "9.0" ]; then
+      CUDNN_VERSION_DETAILED="7.5.0.56"
+    elif [ "$CUDA_VERSION" = "9.2" ]; then
+      CUDNN_VERSION_DETAILED="7.5.0.56"
+    elif [ "$CUDA_VERSION" = "10.0" ]; then
+      CUDNN_VERSION_DETAILED="7.5.0.56"
+    elif [ "$CUDA_VERSION" = "10.1" ]; then
+      CUDNN_VERSION_DETAILED="7.5.0.56"
+    elif [ -n "$CUDNN_VERSION" ]; then
+      echo "Error: cuDNN $CUDNN_VERSION is not compatible with Cuda $CUDA_VERSION."
+      return 1
+    fi
+
+  elif [ -n "$CUDNN_VERSION" ]; then
+    echo "Error: You need to set CUDNN_VERSION to 7.0, 7.1, 7.4 or 7.5."
+    return 1
+  fi
+
+  # Setup URLs
+  CUDNN_URL="https://developer.download.nvidia.com/compute/machine-learning/repos/ubuntu1604/x86_64/libcudnn7_${CUDNN_VERSION_DETAILED}-1+cuda${CUDA_VERSION}_amd64.deb"
+  CUDNN_URL_DEV="https://developer.download.nvidia.com/compute/machine-learning/repos/ubuntu1604/x86_64/libcudnn7-dev_${CUDNN_VERSION_DETAILED}-1+cuda${CUDA_VERSION}_amd64.deb"
+
+  # Setup temporary paths
+  CUDNN_TMP_PATH="/tmp/cudnn.deb"
+  CUDNN_DEV_TMP_PATH="/tmp/cudnn-dev.deb"
+
+  CUDNN_TMP_DIR_PATH="/tmp/cudnn"
+  CUDNN_DEV_TMP_DIR_PATH="/tmp/cudnn-dev"
+
+  echo "Download binaries."
+  wget "$CUDNN_URL" -O "$CUDNN_TMP_PATH"
+  wget "$CUDNN_URL_DEV" -O "$CUDNN_DEV_TMP_PATH"
+
+  mkdir -p "$CUDNN_TMP_DIR_PATH"
+  mkdir -p "$CUDNN_DEV_TMP_DIR_PATH"
+
+  echo "Extract binaries."
+  cd "$CUDNN_TMP_DIR_PATH"
+  ar x "$CUDNN_TMP_PATH"
+  tar -xJf data.tar.xz
+  cd "$CUDNN_DEV_TMP_DIR_PATH"
+  ar x "$CUDNN_DEV_TMP_PATH"
+  tar -xJf data.tar.xz
+
+  echo "Install cuDNN files."
+
+  PARENT_BASE_DIR=$(dirname $CUDA_HOME)
+  if [ ! -w "$PARENT_BASE_DIR" ]; then
+    sudo mv $CUDNN_TMP_DIR_PATH/usr/lib/x86_64-linux-gnu/libcudnn* "$CUDA_HOME/lib64/"
+    sudo mv "$CUDNN_DEV_TMP_DIR_PATH/usr/include/x86_64-linux-gnu/cudnn_v7.h" "$CUDA_HOME/include/"
+    sudo mv "$CUDNN_DEV_TMP_DIR_PATH/usr/lib/x86_64-linux-gnu/libcudnn_static_v7.a" "$CUDA_HOME/lib64/"
+
+    sudo rm -f "$CUDA_HOME/include/cudnn.h"
+    sudo rm -f "$CUDA_HOME/lib64/libcudnn_static.a"
+
+    sudo ln -s "$CUDA_HOME/include/cudnn_v7.h" "$CUDA_HOME/include/cudnn.h"
+    sudo ln -s "$CUDA_HOME/lib64/libcudnn_static_v7.a" "$CUDA_HOME/lib64/libcudnn_static.a"
+  else
+    mv $CUDNN_TMP_DIR_PATH/usr/lib/x86_64-linux-gnu/libcudnn* "$CUDA_HOME/lib64/"
+    mv "$CUDNN_DEV_TMP_DIR_PATH/usr/include/x86_64-linux-gnu/cudnn_v7.h" "$CUDA_HOME/include/"
+    mv "$CUDNN_DEV_TMP_DIR_PATH/usr/lib/x86_64-linux-gnu/libcudnn_static_v7.a" "$CUDA_HOME/lib64/"
+
+    rm -f "$CUDA_HOME/include/cudnn.h"
+    rm -f "$CUDA_HOME/lib64/libcudnn_static.a"
+
+    ln -s "$CUDA_HOME/include/cudnn_v7.h" "$CUDA_HOME/include/cudnn.h"
+    ln -s "$CUDA_HOME/lib64/libcudnn_static_v7.a" "$CUDA_HOME/lib64/libcudnn_static.a"
+  fi
+
+  echo "Cleanup files."
+  rm -fr "$CUDNN_TMP_DIR_PATH"
+  rm -fr "$CUDNN_DEV_TMP_DIR_PATH"
+  rm -f "$CUDNN_TMP_PATH"
+  rm -f "$CUDNN_DEV_TMP_PATH"
+
+  echo "cuDNN $CUDNN_VERSION is installed at $CUDA_HOME."
+}
+
+cuda.install.nccl() {
+  # Install NCCL in $CUDA_HOME
+
+  if ! is_cuda_home_set; then
+    return 1
+  fi
+
+  if ! is_cuda_home_symbolic_link; then
+    return 1
+  fi
+
+  CUDA_VERSION="$(guess_cuda_version)"
+  if [ -z "$CUDA_VERSION" ]; then
+    echo "Can't guess the Cuda version from $CUDA_HOME."
+    return 1
+  fi
+
+  NCCL_VERSION="$1"
+  if [ -z "$NCCL_VERSION" ]; then
+    # echo "Please specify a NCCL version."
+    # echo "Usage: cuda.install.nccl NCCL_VERSION"
+    # echo "Example: cuda.install.nccl 2.4"
+    # echo "NCCL version available: 2.1, 2.2, 2.3 and 2.4"
+    # return 1
+    # Default NCCL version
+    NCCL_VERSION="2.4"
+  fi
+
+  # NCCL 2.1
+  if [ "$NCCL_VERSION" = "2.1" ]; then
+
+
+    if [ "$CUDA_VERSION" = "9.0" ]; then
+      NCCL_VERSION_DETAILED="2.1.15-1"
+    elif [ "$CUDA_VERSION" = "9.1" ]; then
+      NCCL_VERSION_DETAILED="2.1.15-1"
+    elif [ -n "$NCCL_VERSION" ]; then
+      echo "Error: NCCL $NCCL_VERSION is not compatible with Cuda $CUDA_VERSION."
+      return 1
+    fi
+
+  # NCCL 2.3
+  elif [ "$NCCL_VERSION" = "2.2" ]; then
+
+    # NCCL 2.2
+    if [ "$CUDA_VERSION" = "9.0" ]; then
+      NCCL_VERSION_DETAILED="2.2.13-1"
+    elif [ "$CUDA_VERSION" = "9.2" ]; then
+      NCCL_VERSION_DETAILED="2.2.13-1"
+    elif [ -n "$NCCL_VERSION" ]; then
+      echo "Error: NCCL $NCCL_VERSION is not compatible with Cuda $CUDA_VERSION."
+      return 1
+    fi
+
+  # NCCL 2.3
+  elif [ "$NCCL_VERSION" = "2.3" ]; then
+
+    if [ "$CUDA_VERSION" = "9.0" ]; then
+      NCCL_VERSION_DETAILED="2.3.7-1"
+    elif [ "$CUDA_VERSION" = "9.2" ]; then
+      NCCL_VERSION_DETAILED="2.3.7-1"
+    elif [ "$CUDA_VERSION" = "10.0" ]; then
+      NCCL_VERSION_DETAILED="2.3.7-1"
+    elif [ -n "$NCCL_VERSION" ]; then
+      echo "Error: NCCL $NCCL_VERSION is not compatible with Cuda $CUDA_VERSION."
+      return 1
+    fi
+
+  # NCCL 2.4
+  elif [ "$NCCL_VERSION" = "2.4" ]; then
+
+    if [ "$CUDA_VERSION" = "9.0" ]; then
+      NCCL_VERSION_DETAILED="2.4.2-1"
+    elif [ "$CUDA_VERSION" = "9.2" ]; then
+      NCCL_VERSION_DETAILED="2.4.2-1"
+    elif [ "$CUDA_VERSION" = "10.0" ]; then
+      NCCL_VERSION_DETAILED="2.4.2-1"
+    elif [ "$CUDA_VERSION" = "10.1" ]; then
+      NCCL_VERSION_DETAILED="2.4.2-1"
+    elif [ -n "$NCCL_VERSION" ]; then
+      echo "Error: NCCL $NCCL_VERSION is not compatible with Cuda $CUDA_VERSION."
+      return 1
+    fi
+
+  elif [ -n "$NCCL_VERSION" ]; then
+    echo "Error: You need to set NCCL_VERSION to 2.1, 2.2, 2.3 and 2.4."
+    return 1
+  fi
+
+  # Setup URLs
+  NCCL_URL="https://developer.download.nvidia.com/compute/machine-learning/repos/ubuntu1604/x86_64/libnccl2_${NCCL_VERSION_DETAILED}+cuda${CUDA_VERSION}_amd64.deb"
+  NCCL_URL_DEV="https://developer.download.nvidia.com/compute/machine-learning/repos/ubuntu1604/x86_64/libnccl-dev_${NCCL_VERSION_DETAILED}+cuda${CUDA_VERSION}_amd64.deb"
+
+  # Setup temporary paths
+  NCCL_TMP_PATH="/tmp/nccl.deb"
+  NCCL_DEV_TMP_PATH="/tmp/nccl-dev.deb"
+
+  NCCL_TMP_DIR_PATH="/tmp/nccl"
+  NCCL_DEV_TMP_DIR_PATH="/tmp/nccl-dev"
+
+  echo "Download binaries."
+  wget "$NCCL_URL" -O "$NCCL_TMP_PATH"
+  wget "$NCCL_URL_DEV" -O "$NCCL_DEV_TMP_PATH"
+
+  mkdir -p "$NCCL_TMP_DIR_PATH"
+  mkdir -p "$NCCL_DEV_TMP_DIR_PATH"
+
+  echo "Extract binaries."
+  cd "$NCCL_TMP_DIR_PATH"
+  ar x "$NCCL_TMP_PATH"
+  tar -xJf data.tar.xz
+  cd "$NCCL_DEV_TMP_DIR_PATH"
+  ar x "$NCCL_DEV_TMP_PATH"
+  tar -xJf data.tar.xz
+
+  echo "Install NCCL files."
+
+  PARENT_BASE_DIR=$(dirname $CUDA_HOME)
+  if [ ! -w "$PARENT_BASE_DIR" ]; then
+    sudo mv $NCCL_TMP_DIR_PATH/usr/lib/x86_64-linux-gnu/libnccl* "$CUDA_HOME/lib64/"
+    sudo rm -f "$CUDA_HOME/include/nccl.h"
+    sudo mv "$NCCL_DEV_TMP_DIR_PATH/usr/include/nccl.h" "$CUDA_HOME/include/nccl.h"
+    sudo rm -f "$CUDA_HOME/lib64/libnccl_static.a"
+    sudo mv "$NCCL_DEV_TMP_DIR_PATH/usr/lib/x86_64-linux-gnu/libnccl_static.a" "$CUDA_HOME/lib64/libnccl_static.a"
+  else
+    mv $NCCL_TMP_DIR_PATH/usr/lib/x86_64-linux-gnu/libnccl* "$CUDA_HOME/lib64/"
+    rm -f "$CUDA_HOME/include/nccl.h"
+    mv "$NCCL_DEV_TMP_DIR_PATH/usr/include/nccl.h" "$CUDA_HOME/include/nccl.h"
+    rm -f "$CUDA_HOME/lib64/libnccl_static.a"
+    mv "$NCCL_DEV_TMP_DIR_PATH/usr/lib/x86_64-linux-gnu/libnccl_static.a" "$CUDA_HOME/lib64/libnccl_static.a"
+  fi
+
+  echo "Cleanup files."
+  rm -fr "$NCCL_TMP_DIR_PATH"
+  rm -fr "$NCCL_DEV_TMP_DIR_PATH"
+  rm -f "$NCCL_TMP_PATH"
+  rm -f "$NCCL_DEV_TMP_PATH"
+
+  echo "NCCL $NCCL_VERSION is installed at $CUDA_HOME."
+}
+
+cuda.gcc.install() {
+
+  if [ -z "$1" ]; then
+    echo "Please specify a GCC version."
+    return
+  fi
+  export GCC_VERSION="$1"
+
+  sudo apt install --yes gcc-$GCC_VERSION g++-$GCC_VERSION
+
+  sudo update-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-$GCC_VERSION 10
+  sudo update-alternatives --install /usr/bin/g++ g++ /usr/bin/g++-$GCC_VERSION 10
+
+  sudo update-alternatives --set gcc "/usr/bin/gcc-$GCC_VERSION"
+  sudo update-alternatives --set g++ "/usr/bin/g++-$GCC_VERSION"
+}
diff --git a/tensorflow/ubuntu-18.04/docker-compose.yml b/tensorflow/ubuntu-18.04/docker-compose.yml
new file mode 100755
index 0000000..3c383ea
--- /dev/null
+++ b/tensorflow/ubuntu-18.04/docker-compose.yml
@@ -0,0 +1,16 @@
+version: '3'
+services:
+  tf:
+    build: .
+    stdin_open: true
+    tty: true
+    volumes:
+      - ../../wheels:/wheels
+    environment:
+      - TF_VERSION_GIT_TAG=${TF_VERSION_GIT_TAG:?TF_VERSION_GIT_TAG}
+      - PYTHON_VERSION=${PYTHON_VERSION:?PYTHON_VERSION}
+      - BAZEL_VERSION=${BAZEL_VERSION:?BAZEL_VERSION}
+      - USE_GPU=${USE_GPU-0}
+      - CUDA_VERSION=${CUDA_VERSION-10.0}
+      - CUDNN_VERSION=${CUDNN_VERSION-7.5}
+      - NCCL_VERSION=${NCCL_VERSION-2.4}

From 61153a5674ba50c4edb66368ab808619e6fe2b59 Mon Sep 17 00:00:00 2001
From: Jeremy Fix <jeremy.fix@centralesupelec.fr>
Date: Fri, 22 Nov 2019 10:59:12 +0100
Subject: [PATCH 2/7] rm useless swp file

---
 tensorflow/ubuntu-18.04/.build.sh.swp | Bin 16384 -> 0 bytes
 1 file changed, 0 insertions(+), 0 deletions(-)
 delete mode 100644 tensorflow/ubuntu-18.04/.build.sh.swp

diff --git a/tensorflow/ubuntu-18.04/.build.sh.swp b/tensorflow/ubuntu-18.04/.build.sh.swp
deleted file mode 100644
index 98bbf89f3f6d117c7f1fa918f99f29ea2d748449..0000000000000000000000000000000000000000
GIT binary patch
literal 0
HcmV?d00001

literal 16384
zcmeI3UyS2M9mnSuN=yG-d!<MqUZ&ol?&)y6d)M9`+Xm!KyxDZKNl3E2F73%U&LnZy
zv5oC)ZhKk~e<YrG1O$EHp>PsBpcV<F0s;wm0Kp@o{F8Wr@<8DMA;c5nJN~n=v&-I9
z;sMdI^pWH7%x`AC^P4{&JH6$nYTNWmahc$FfROUbuU`A}!ne=-zCp;qx~@ORe17ND
zBM4pFX3nW`_|1<c29;{9tMpxSkNI-QoWS)5wmX%36U**1Uzr6dlitJ$C-Mr)_ox!x
zF9xGq<|85q1%v`~3QWR*eC0lJ>EcS%)fc6Q={p|WnTsSYg#tnWp@2|8C?FIN3J3*+
z0z!fRX9YrUfqWblTu3T;JvpD5JHI8lzArf+%-v5e#Yre26c7ps1%v`Z0il3UKqw#-
z5DEwdgaSf=|3L+eJ|Xv`T=(z|0KETCpZ{M+;2*&s!1us+!3$s?48RU}4|oWi0}p_I
zoF(Kn@Fmay4}src!{F!OXW+|V3O)c9!C&qr<U8OM@D=bPFu~p62lo*2eeepn2{wQR
zR=|5f0h|SYx|@*CfC11z1zZ5{1O)uz?XUxGf+1*u_X7#s3I28$A-@Jc1}}jxf=`1X
z(7-Zy7gzviz+dkq<VWC#;1KMBr@<9a0RMU$A+LgOfKP!v&;(0h0sQU`LVgQg1kZvG
zfd<$H9|V`cMerav2mbz6LVgLp1wIE{@Fchk{1lrqKLK9{CGZ4T1((6Q!9UTazXIO`
zp9NdsUclS>4lsYv+@WMY#0Xv24iwMwbk8vNj3HZ#zQ)iqAtaZY%CcdGmh0%u*)K`;
zie9g+x7GHp-cq}prNWK5kVA<+d(6Vf4MN?voITeIOOkAlZzqx;o)qg=Rx<Zf13lw;
zX49yisEz4518Z19Nl;M2?0^xJx3jap(X3arw%$;;wUSi0(cRFW(%PL`v!UZN7rm=i
z4`uJ5aHF-`-9%V&?@%@;eItrp-l?d$V8=<r8?*QlGHa>jEw!q}S<SwZk%!)dmnD0=
z=^HOA#!=&O%_BEEt!b?ReTGVfosOngTRRe!*+;3gBGGG)({RMUaw;mZFHC$!7HQcX
zdzNi*SLa_$5lZcTd>=pAb<g!fI_K+flc>?V&1QWIKEiv1Vq3k-kFK52FJNYocF`3W
z4W}-Jqq2Y+KBLx{8#kPg4qTs?jo;g|oFNTD<`ub1B>6YW<+@&Lbh>K2p7g({0GrKi
z?TyBQe@EsLM=lkzS2JTgnx<9suGZ)@+wE>?`S?9vhSEwd96>Xuk9C?|^FEN&2<D?2
zq@_p{O)fJLV}rM>dt@xz9;LX65{1yr+;u#|H^wYven4*NB}qZcWxd(z>Kk>n+QASu
zHhgnba^NoeOoI1pGBo6YJE5lGP}a9Xw4)W!eZ#kmo{biFX~05iI0qCXk2SWQk321i
z(oE6zXyzoRDze&acDj1Gx!tPOF(kz!?b38)nIk$h&GSd@%KXTj>jX#1(*|od%iCzI
zna)u&)~bzKS34fAUhU)pJ6cy)8<lpmR?)9E+glxsStlfGwzLLZva?&R=aNTxRV(e|
zfm`av7I&7pa&;~p4@2#9=_}2vje1k9=-XQ-<Tf986uVqk8`Yel$VrX|M}>yxt>>lC
z8tQso)1R)ZIU`Z*C$(*K#b&v*!aH|^u|99<9;1CL;O*Y0u0zYr37Z{yxp;ZL*0FrY
z&2CmUIypV!62l*Mx|&{ATk3kPUhCGhPU*4Y#rIt*J~nT;T5jbGM5X0ckXdf5%l3uT
zNHWE#?JDMX_*1*t>>l!gEgvCS0Q~0&e5nvSuT)qnBv#jJjVO%DCVeCx`e<N<>~ZS(
zmJ=@V-^C#d`8ipm1WV_iS-y7uk@Hf%0D5YxR7g69cMy(Thxd<G^b55MN3$+s_Guw0
zE=kU%%<2fMMRZ)#b{&R=LZ5j^yMkXm-*C|Lc+zBGN|yvw8ik=3tSJgojwZdL>5el7
zGe3)nAQ68|<pIX{qmN}3f-)&sM$-;sG&MGzPK&cRMb{rHmJ@`AZKr2Fa3{Wr34~7>
zV;6%@pM{2H2Su3LO~xU^4qVJN%-D-&mHfbCIQ?jNawkE?2acwoc)5h?q>~~yoEzoL
zk>Lz6vSGgJM)ldi$G>zRgDR%bqj4?j5B!5*3~;IAWfKsBr6)~lNAV@G5y*z;*_N3M
zka3XbGtYNX;Q;d@!Mu!7JWV9a2N1%UjJ*Ro9WlPv<14;Wsj#koSgWUPMq<e*U(W6u
zAxeJK3W-6}XFNQ%Cp(!4G<GbQ%rF1agwdK335|ZAa$Ov(D``|p84Wb#>x)|4fS3X(
z>gT)gcAi<{r&wHDzUz!JRKN>O6Sfj1$;9g;qHNfhui^<Oz+FC@3@==uk}?T=rDr(`
zuH*urJdP&MQ&J*_OiI(dt(2+Y*({?%WCL=RC_Yj>xfI83p-8;qGh1_|;<hr?D@m(-
z;_W~O)AfC3h7@;G|4Wu|QRkwRuNRAHj`4yZ5sC7v*~)~@uYwYvi45S2bxIA}G6EV*
z`ff?eFHcm=+WgTWs(|g9hOO{q(rqs>qVZJXi_*TqTXZdzVy(b$Nrdo5m*Gqgr*{<i
z4ZIyC<In%+&Jtn({``-}^5}RC&+}gcH^EoIC&4Gc{opKk9r3>aJ`bjVr~e2LhfqK$
zAQTV^2nB=!LII(GP(Uak6c7r$Nfj_hW4BM`HeFp^rAoBV5-4~8->|MHyDkHl|LvQ4
z%pY6Wr-?SD7vsGcnQu_c_FI%=uOZ_<g=b$wVt*pZq>z6($DjIl_)8RQJRwVLI`LN}
zd>^d4L7ynmmEsjbq8*_BYOdIG!LH47*k{lkygb00rR-}wJe-gB`j$^+@05ba@0Ie^
Nc?0hnPkDDp{tY;HQ8@qr


From 59b378139e03a908c3a08c1c7d84688e26b73a6f Mon Sep 17 00:00:00 2001
From: Jeremy Fix <jeremy.fix@centralesupelec.fr>
Date: Fri, 22 Nov 2019 11:01:23 +0100
Subject: [PATCH 3/7] adds cuda_capabilities 6.1 for the 1080Ti

---
 tensorflow/ubuntu-18.04/build.sh | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tensorflow/ubuntu-18.04/build.sh b/tensorflow/ubuntu-18.04/build.sh
index 02f926f..62c0f7e 100755
--- a/tensorflow/ubuntu-18.04/build.sh
+++ b/tensorflow/ubuntu-18.04/build.sh
@@ -48,7 +48,7 @@ export PYTHON_ARG=${TF_ROOT}/lib
 # Compilation parameters
 export TF_NEED_CUDA=0
 export TF_NEED_GCP=1
-export TF_CUDA_COMPUTE_CAPABILITIES=5.2,3.5
+export TF_CUDA_COMPUTE_CAPABILITIES=6.1,5.2,3.5
 export TF_NEED_HDFS=1
 export TF_NEED_OPENCL=0
 export TF_NEED_JEMALLOC=1  # Need to be disabled on CentOS 6.6

From 30cc6dbc993dd61a41df0eb21eec14bba4797c02 Mon Sep 17 00:00:00 2001
From: Jeremy Fix <jeremy.fix@centralesupelec.fr>
Date: Fri, 22 Nov 2019 11:01:43 +0100
Subject: [PATCH 4/7] set -ex to printing the executed commands

---
 tensorflow/ubuntu-18.04/build.sh | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tensorflow/ubuntu-18.04/build.sh b/tensorflow/ubuntu-18.04/build.sh
index 62c0f7e..3ab8b72 100755
--- a/tensorflow/ubuntu-18.04/build.sh
+++ b/tensorflow/ubuntu-18.04/build.sh
@@ -1,5 +1,5 @@
 #!/usr/bin/env bash
-set -e
+set -ex
 
 export PATH="/conda/bin:/usr/bin:$PATH"
 

From 56691f3ad1bb1a370716ff2cb490031daab984b2 Mon Sep 17 00:00:00 2001
From: Jeremy Fix <jeremy.fix@centralesupelec.fr>
Date: Fri, 22 Nov 2019 11:03:36 +0100
Subject: [PATCH 5/7] adds the link to the cuda capabilities in the README

---
 README.md | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/README.md b/README.md
index d7b9167..e4cdf29 100755
--- a/README.md
+++ b/README.md
@@ -24,7 +24,8 @@ git clone https://github.com/hadim/docker-tensorflow-builder.git
 - Edit the `build.sh` file to modify TensorFlow compilation parameters. Then launch the build:
 
 ```bash
-LINUX_DISTRO="ubuntu-16.04"
+# LINUX_DISTRO="ubuntu-16.04"
+LINUX_DISTRO="ubuntu-18.04"
 # or LINUX_DISTRO="ubuntu-18.10"
 # or LINUX_DISTRO="centos-7.4"
 # or LINUX_DISTRO="centos-6.6"
@@ -52,7 +53,8 @@ docker-compose run tf
 - Edit the `build.sh` file to modify TensorFlow compilation parameters. Then launch the build:
 
 ```bash
-LINUX_DISTRO="ubuntu-16.04"
+# LINUX_DISTRO="ubuntu-16.04"
+LINUX_DISTRO="ubuntu-18.04"
 # or LINUX_DISTRO="ubuntu-18.10"
 # or LINUX_DISTRO="centos-7.4"
 # or LINUX_DISTRO="centos-6.6"
@@ -81,6 +83,7 @@ docker-compose run tf
 ---
 
 - Refer to [tested build configurations](https://www.tensorflow.org/install/source#tested_build_configurations) to know which `BAZEL_VERSION` you need.
+- Refer to [the nvidia cuda capabilites table](https://developer.nvidia.com/cuda-gpus#compute) to check the cuda capabilities of your GPU and set accordingly the variable `TF_CUDA_COMPUTE_CAPABILITIES` within the `build.sh` script. 
 - Be patient, the compilation can be long.
 - Enjoy your Python wheels in the `wheels/` folder.
 - *Don't forget to remove the container to free the space after the build: `docker-compose rm --force`.*

From fa2adbacb50b395bea2aa9f7109e034e857ec87d Mon Sep 17 00:00:00 2001
From: Jeremy Fix <jeremy.fix@centralesupelec.fr>
Date: Fri, 22 Nov 2019 11:04:11 +0100
Subject: [PATCH 6/7] corrects the nvidia download link to the cuda run file
 for 10.1

---
 tensorflow/ubuntu-18.04/cuda.sh | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tensorflow/ubuntu-18.04/cuda.sh b/tensorflow/ubuntu-18.04/cuda.sh
index 8086b01..4c20c52 100755
--- a/tensorflow/ubuntu-18.04/cuda.sh
+++ b/tensorflow/ubuntu-18.04/cuda.sh
@@ -131,7 +131,7 @@ cuda.install.cuda() {
   elif [ "$CUDA_VERSION" = "10.0" ]; then
     CUDA_URL="https://developer.nvidia.com/compute/cuda/10.0/Prod/local_installers/cuda_10.0.130_410.48_linux"
   elif [ "$CUDA_VERSION" = "10.1" ]; then
-    CUDA_URL="https://developer.nvidia.com/compute/cuda/10.1/Prod/local_installers/cuda_10.1.105_418.39_linux.run"
+    CUDA_URL="http://developer.download.nvidia.com/compute/cuda/10.1/Prod/local_installers/cuda_10.1.243_418.87.00_linux.run"
   else
     echo "Error: You need to set CUDA_VERSION to 9.0, 9.1, 9.2, 10.0 or 10.1."
     return 1

From 916a423b09dab0dda5164113134237508813b4eb Mon Sep 17 00:00:00 2001
From: Jeremy Fix <jeremy.fix@centralesupelec.fr>
Date: Fri, 22 Nov 2019 11:05:20 +0100
Subject: [PATCH 7/7] use a 18.04 docker base image

---
 tensorflow/ubuntu-18.04/Dockerfile | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tensorflow/ubuntu-18.04/Dockerfile b/tensorflow/ubuntu-18.04/Dockerfile
index 538bb43..62f8d39 100755
--- a/tensorflow/ubuntu-18.04/Dockerfile
+++ b/tensorflow/ubuntu-18.04/Dockerfile
@@ -1,4 +1,4 @@
-FROM ubuntu:18.10
+FROM ubuntu:18.04
 
 RUN apt update && apt install -y \
     build-essential \