backport to other distros

Author: hadim

tensorflow/centos-6.6/Dockerfile

@@ -27,6 +27,6 @@ RUN wget "https://repo.continuum.io/miniconda/Miniconda3-latest-Linux-x86_64.sh"
 
 COPY build.sh /build.sh
 COPY build2.sh /build2.sh
-COPY setup_cuda.sh /setup_cuda.sh
+COPY cuda.sh /cuda.sh
 
 CMD bash build.sh

tensorflow/centos-6.6/build.sh

@@ -4,7 +4,11 @@ set -e
 source /root/.bashrc
 
 if [ "$USE_GPU" -eq "1" ]; then
-    bash setup_cuda.sh
+  export CUDA_HOME="/usr/local/cuda"
+  alias sudo=""
+  source cuda.sh
+  cuda.install $CUDA_VERSION $CUDNN_VERSION $NCCL_VERSION
+  cd /
 fi
 
 # Enable GCC 6

tensorflow/centos-6.6/build2.sh

@@ -4,10 +4,11 @@ set -ex
 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
-conda install --yes -c conda-forge keras-applications keras-preprocessing
+pip install keras-applications keras-preprocessing
 
 # Compile TensorFlow