#!/bin/bash
  
  # Installs chainer with nn-gev dependencies
  # miniconda should be installed in $HOME/miniconda3/ 
  # Download cuDNN from "https://developer.nvidia.com/rdp/cudnn-download" and extract in "$HOME/cuda/"
  # and add their paths "export CFLAGS=-I$HOME/cuda/include" "export LDFLAGS=-L$HOME/cuda/lib64"
  
  cudnn_dir=$HOME/cuda
  cudnn_include_file=$cudnn_dir/include/cudnn.h
  cudnn_lib_dir=$cudnn_dir/lib64
  miniconda_dir=$HOME/miniconda3/
  
  if [ ! -d $miniconda_dir ]; then
      echo "$miniconda_dir does not exist. Please run 'tools/extras/install_miniconda.sh" && exit 1;
  fi
  
  if [ ! -d $cudnn_lib_dir ] || [ ! -f $cudnn_include_file ]; then
      echo "cuDNN is not available. $cudnn_include_file and/or $cudnn_lib_dir are missing.
            Download cuDNN v5.1 for appropriate CUDA version (7.5 or 8.0) from 'https://developer.nvidia.com/rdp/cudnn-download'.
  	  Check CUDA version using the command 'nvcc --version'
  	  Place the include and lib directories in $cudnn_dir after download" && exit 1;
  fi
  
  cudnn_major=`cat $HOME/cuda/include/cudnn.h | grep CUDNN_MAJOR | head -1 | rev | cut -d " " -f1`
  cudnn_minor=`cat $HOME/cuda/include/cudnn.h | grep CUDNN_MINOR | head -1 | rev | cut -d " " -f1`
  
  if [ $cudnn_major -ne 5 ] || [ $cudnn_minor -ne 1 ]; then
      echo "cuDNN version in $cudnn_dir is not '5.1'. Please download v5.1"  && exit 1;
  fi
  
  $HOME/miniconda3/bin/python -m pip install --user chainer==1.16.0