nqg icon indicating copy to clipboard operation
nqg copied to clipboard

Can we run it on CPU machine (No GPU)

Open karam86 opened this issue 6 years ago • 3 comments

I don't have GPU machine. When I try to run the testing using the existing trained model. It given me error "cutorch" not installed. And for cutorch I need CUDA and for CUDA I need gpu machine. Is there a way around??

karam86 avatar Mar 23 '18 16:03 karam86

You can solve this problem by manually modifying the config-trans file in the model dir. Simply:

src=../data/processed/src-dev.txt
beam_size=3
fallback_to_cpu=true replace_unk=true

then you can run prediction on your CPU machine. Hope this would help you.

Aria-K-Alethia avatar May 09 '18 00:05 Aria-K-Alethia

Remove "gpuid=1" in the config-trans (for testing) and config-train (for training) files. After this change it works. @karam86 @Aria-K-Alethia

saket-maheshwary avatar Jun 21 '18 21:06 saket-maheshwary

Indeed, or change the gpuid to 0 and it will use CPU.

TjebbeVQ avatar Jan 30 '19 19:01 TjebbeVQ