R9868/install_pyeddl_GPU.sh
0614822bd28bpyeddl_1.0.0

install_pyeddl_GPU.sh
View Options

			#!/bin/sh

			git clone --recurse-submodules https://github.com/deephealthproject/pyeddl.git

			EIGEN_PATH=/home/andrea/anaconda3/envs/eddl/include/eigen3
			export EDDL_WITH_PROTOBUF=OFF

			export EDDL_DIR=/home/$USER/eddl
			export CPATH="${EIGEN_PATH}:${CPATH}"
			cd pyeddl/
			PYEDDL_FOLDER=$(pwd)
			cd third_party/eddl

			# Compiling eddl
			mkdir build
			cd build
			cmake .. -DBUILD_TARGET=GPU -DBUILD_SHARED_LIB=ON -DMKL=ON -DBUILD_PROTOBUF=$EDDL_WITH_PROTOBUF -DBUILD_OPENMP=ON -DBUILD_HPC=ON -DBUILD_TESTS=ON -DBUILD_EXAMPLES=ON -DCMAKE_INSTALL_PREFIX=$EDDL_DIR
			make -j$(nproc)

			# Moving relevant files in /home/$USER/eddl/include and /home/$USER/eddl/lib
			make install

			# Must export the EDDL_WITH_CUDA environment variable
			export EDDL_WITH_CUDA="true"
			#ln -s /usr/local/cuda-10.1/targets/x86_64-linux/lib/libcudart.so /usr/lib/
			#ln -s /usr/local/cuda-10.1/targets/x86_64-linux/lib/libcurand.so /usr/lib/
			#ln -s /usr/local/cuda-10.1/targets/x86_64-linux/lib/libcublas.so /usr/lib/

			# Installing required packages for pyeddl
			conda install -y -n eddl pybind11 pytest

			# Installing pyeddl
			cd $PYEDDL_FOLDER
			python3 setup.py install