install_pyeddl_GPU.sh
4c8a36152fa6master

install_pyeddl_GPU.sh
View Options

			#!/bin/sh

			git clone --recurse-submodules https://github.com/deephealthproject/pyeddl.git
			cd pyeddl/
			git fetch --tags
			git checkout tags/0.6.0
			git submodule update --recursive

			PYEDDL_FOLDER=$(pwd)
			cd third_party/eddl

			EIGEN_PATH=/scrap/users/$USER/miniconda3/envs/eddl/include/eigen3
			export EDDL_WITH_PROTOBUF=OFF

			export EDDL_DIR=/scrap/users/$USER/eddl
			export CPATH="${EIGEN_PATH}:${CPATH}"

			# Compiling eddl
			mkdir build
			cd build
			cmake3 .. -DEDDL_SHARED=ON -DBUILD_TARGET=CPU -DCMAKE_CUDA_COMPILER=/usr/local/cuda-10.1/bin/nvcc -DBUILD_PROTOBUF=$EDDL_WITH_PROTOBUF -DBUILD_EXAMPLES=OFF -DBUILD_TESTS=OFF -DCMAKE_INSTALL_PREFIX=$EDDL_DIR
			make -j$(nproc)

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

			cd $PYEDDL_FOLDER
			python setup.py install