Commit df377973 authored by Pratyush Das's avatar Pratyush Das

FIX: Set parameters for Tensorflow

Add keras_applications, keras_preprocessing
Add LD_Library Path
Enable XLA/JIT
parent 918e5d46
......@@ -82,7 +82,7 @@ RUN apt-get update -y \
# ----------------------------------------------------------------------------
RUN ln -s /usr/bin/python3 /usr/bin/python \
&& python3 -m pip install --upgrade pip \
&& python3 -m pip install pip six numpy wheel mock keras future setuptools
&& python3 -m pip install pip six numpy wheel mock keras keras_applications keras_preprocessing future setuptools
# ----------------------------------------------------------------------------
# Build TensorFlow
......@@ -100,6 +100,7 @@ RUN export TF_ROOT=/work/tf \
&& git clone https://github.com/tensorflow/tensorflow.git \
&& cd tensorflow \
&& git checkout r2.1 \
&& export LD_LIBRARY_PATH="$LD_LIBRARY_PATH:/usr/local/cuda/lib64/:/usr/local/cuda/lib64/stubs/" \
&& export PYTHON_BIN_PATH=$(which python) \
&& export PYTHON_LIB_PATH="$($PYTHON_BIN_PATH -c 'import site; print(site.getsitepackages()[0])')" \
&& export PYTHONPATH=/usr/lib/ \
......@@ -109,11 +110,11 @@ RUN export TF_ROOT=/work/tf \
&& export TF_NEED_GCP=0 \
&& export TF_NEED_CUDA=1 \
&& export TF_CUDA_VERSION="$($CUDA_TOOLKIT_PATH/bin/nvcc --version | sed -n 's/^.*release \(.*\),.*/\1/p')" \
&& export TF_CUDA_COMPUTE_CAPABILITIES=6.1,5.2,3.5 \
&& export TF_CUDA_COMPUTE_CAPABILITIES=7.5,6.1,5.2 \
&& export TF_NEED_HDFS=0 \
&& export TF_NEED_OPENCL=0 \
&& export TF_NEED_JEMALLOC=1 \
&& export TF_ENABLE_XLA=0 \
&& export TF_ENABLE_XLA=1 \
&& export TF_NEED_VERBS=0 \
&& export TF_CUDA_CLANG=0 \
&& export TF_CUDNN_VERSION="$(sed -n 's/^#define CUDNN_MAJOR\s*\(.*\).*/\1/p' $CUDNN_INSTALL_PATH/include/cudnn.h)" \
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment