install_pyeddl_GPU.sh
No OneTemporary
Actions

Subscribers

None

File Metadata

	#!/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