Add CUDA_DIR; disable CUDNN by default; remove nematus

1 files changed, 5 insertions, 4 deletions

diff --git a/Makefile b/Makefile
index c8af791..ab3c583 100644
--- a/Makefile
+++ b/Makefile
@@ -4,10 +4,12 @@ GIT_MARIAN_DEV=http://github.com/marian-nmt/marian-dev.git
 GIT_MARIAN=http://github.com/marian-nmt/marian.git
 GIT_MOSES_SCRIPTS=http://github.com/marian-nmt/moses-scripts.git
 GIT_SUBWORD_NMT=http://github.com/rsennrich/subword-nmt.git
-GIT_NEMATUS=http://github.com/EdinburghNLP/nematus.git
 
 BRANCH=master
-USE_CUDNN=ON
+CUDA_DIR=/usr/local/cuda-9.2
+CUDNN=off
+
+CMAKE_FLAGS=-DCUDA_TOOLKIT_ROOT_DIR=$(CUDA_DIR) -DUSE_CUDNN=$(CUDNN)
 
 PIP_PACKAGES=websocket-client pyyaml
 
@@ -25,13 +27,12 @@ run: install
 tools:
 	git -C $@/moses-scripts pull || git clone $(GIT_MOSES_SCRIPTS) $@/moses-scripts
 	git -C $@/subword-nmt pull || git clone $(GIT_SUBWORD_NMT) $@/subword-nmt
-	git -C $@/nematus pull || git clone $(GIT_NEMATUS) $@/nematus
 	pip3 install --user $(PIP_PACKAGES)
 
 tools/marian:
 	git -C $@ pull || git clone $(GIT_MARIAN_DEV) -b $(BRANCH) $@
 	rm -rf $@/build
-	mkdir -p $@/build && cd $@/build && cmake .. -DCOMPILE_EXAMPLES=ON -DUSE_CUDNN=$(USE_CUDNN) && make -j$(THREADS)
+	mkdir -p $@/build && cd $@/build && cmake .. -DCOMPILE_EXAMPLES=ON $(CMAKE_FLAGS) && make -j$(THREADS)
 
 models:
 	mkdir -p $@