Commit 90d36737 authored by Andreas Gocht's avatar Andreas Gocht

update compilers

parent e09d3a0e
......@@ -34,7 +34,8 @@ Most likely you still might want to adapt TF_CUDA_COMPUTE_CAPABILITIES to your d
###################################
apt-get -y install vim
apt-get -y install wget curl build-essential make g++ gfortran git subversion cmake python python3 python-pip python3-pip git
apt-get -y install wget curl build-essential make g++-6 gcc-6 gfortran git subversion cmake python python3 python-pip python3-pip git
apt-get -y remove gcc-7
###################################
# environment
......@@ -66,25 +67,35 @@ Most likely you still might want to adapt TF_CUDA_COMPUTE_CAPABILITIES to your d
apt-get -y install nvidia-cuda-toolkit=9.1.85-3ubuntu1
apt-get -y install nvidia-cuda-dev=9.1.85-3ubuntu1
apt-get -y install libcupti-dev=9.1.85-3ubuntu1
cd /apps
if [ ! -d tensorflow-dep ]; then
mkdir tensorflow-dep
cd tensorflow-dep
cp /tmp/cudnn-9.1-linux-x64-v7.1.tgz ./
tar -vxf cudnn-9.1-linux-x64-v7.1.tgz
cp /tmp/nccl_2.1.15-1+cuda9.1_x86_64.txz ./
tar -xf nccl_2.1.15-1+cuda9.1_x86_64.txz
fi
cd tensorflow-dep
cp /tmp/cudnn-9.1-linux-x64-v7.1.tgz ./
tar -vxf cudnn-9.1-linux-x64-v7.1.tgz
cp /tmp/nccl_2.1.15-1+cuda9.1_x86_64.txz ./
tar -xf nccl_2.1.15-1+cuda9.1_x86_64.txz
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/apps/tensorflow-dep/cuda/lib64:/apps/tensorflow-dep/nccl_2.1.15-1+cuda9.1_x86_64/lib
###################################
# Bazel
###################################
apt-get -y install openjdk-8-jdk
echo "deb [arch=amd64] http://storage.googleapis.com/bazel-apt stable jdk1.8" | tee /etc/apt/sources.list.d/bazel.list
curl https://bazel.build/bazel-release.pub.gpg | apt-key add -
apt-get -y install openjdk-8-jdk bash-completion unzip g++-6 g++
if [ ! -d tensorflow-dep ]; then
mkdir tensorflow-dep
fi
cd tensorflow-dep
wget http://storage.googleapis.com/bazel-apt/pool/jdk1.8/b/bazel/bazel_0.18.1_amd64.deb
dpkg --install bazel_0.18.1_amd64.deb
apt-get -y update
apt-get -y install bazel
......@@ -131,8 +142,11 @@ Most likely you still might want to adapt TF_CUDA_COMPUTE_CAPABILITIES to your d
export TF_CUDNN_VERSION="7.1.2"
export TF_NEED_MKL=1
export TF_DOWNLOAD_MKL=1
export GCC_HOST_COMPILER_PATH=/usr/bin/x86_64-linux-gnu-gcc-6
export HOST_CXX_COMPILER=/usr/bin/x86_64-linux-gnu-g++-6
export HOST_C_COMPILER=/usr/bin/x86_64-linux-gnu-gcc-6
./configure -march=native
./configure
###################################
# Build TensorFlow with MKL
......@@ -157,7 +171,7 @@ Most likely you still might want to adapt TF_CUDA_COMPUTE_CAPABILITIES to your d
mkdir -p /apps/tf_cuda
mkdir -p /apps/tf_mkl
cp /tmp/tensorflow_pkg_mkl/*.whl /apps/tf_cuda
cp /tmp/tensorflow_pkg_cuda/*.whl /apps/tf_cuda
cp /tmp/tensorflow_pkg_mkl/*.whl /apps/tf_mkl
......
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