#!/usr/bin/env bash

###############################################################################
# Copyright 2018 The Apollo Authors. All Rights Reserved.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
###############################################################################

# Fail on first error.
set -e

cd "$(dirname "${BASH_SOURCE[0]}")"

apt-get update -y && apt-get install -y \
    libatlas-base-dev \
    libflann-dev \
    libhdf5-serial-dev \
    libicu-dev \
    liblmdb-dev \
    libopenblas-dev \
    libopencv-dev \
    libopenni-dev \
    libqhull-dev \
    libsnappy-dev \
    libvtk5-dev \
    libvtk5-qt4-dev \
    mpi-default-dev

wget https://apollocache.blob.core.windows.net/apollo-docker/caffe_x86.tar.gz
tar xzf caffe_x86.tar.gz
mv caffe_x86/output-GPU/include/caffe /usr/include/
mv caffe_x86/output-GPU/lib/* /usr/lib/x86_64-linux-gnu/

# Clean up.
apt-get clean && rm -rf /var/lib/apt/lists/*
rm -fr caffe_x86.tar.gz caffe_x86
