diff options
| -rw-r--r-- | Makefile | 9 |
1 files changed, 5 insertions, 4 deletions
@@ -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 $@ |