Commit 9275c858 authored by Andreas Gocht's avatar Andreas Gocht

small fix in the path, and do not build tf at buildtime

parent 321e3aea
......@@ -182,14 +182,16 @@ Namespace: amd64
tar -xf cudnn-9.0-linux-x64-v7.1.tgz
fi
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/extras/CUPTI/lib64
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/extras/CUPTI/lib64:/apps/tensorflow-dep/cuda/lib64
cd /apps
if [ ! -d tensorflow ]; then
git clone https://github.com/tensorflow/tensorflow
fi
cd tensorflow
#TF
cd tensorflow
git checkout v1.6.0
export TF_NEED_CUDA=1
......@@ -209,13 +211,16 @@ Namespace: amd64
export TF_ENABLE_XLA=0
export TF_NEED_VERBS=0
export TF_CUDA_CLANG=0
export TF_CUDNN_VERSION="7.1"
export TF_CUDNN_VERSION="7.1.1"
export TF_NEED_MKL=0
export TF_DOWNLOAD_MKL=0
./configure
bazel build --cxxopt="-D_GLIBCXX_USE_CXX11_ABI=0" --config=opt --config=cuda //tensorflow/tools/pip_package:build_pip_package
#bazel build --cxxopt="-D_GLIBCXX_USE_CXX11_ABI=0" --config=opt --config=cuda //tensorflow/tools/pip_package:build_pip_package
chmod -R a+rwX ./
rm -r bazel-bin bazel-genfiles bazel-tensorflow bazel-out bazel-testlogs
#pip3 install /tmp/tensorflow_pkg/*.whl
......@@ -227,6 +232,10 @@ Namespace: amd64
# TODO CUDNN
#/apps/bazel/output/bazel
%environment
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/extras/CUPTI/lib64:/apps/tensorflow-dep/cuda/lib64
export PATH=$PATH:/apps/bazel/output/
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