Rasa core use multi CPU or GPU

[azuredsky]

How do I use multi CPU or specify one GPU when train rasa core? Will it offer some API?

[amn41 (Alan Nichol)]

Rasa Core’s policies use tensorflow in the backend, so you will have to make sure that you have e.g. CUDA drivers installed and tensorflow configured to run on GPU.

[azuredsky]

@amn41

GPU

I added this config in tensorflow session. “config_proto”: { “device_count”: cpu_count(), “inter_op_parallelism_threads”: 0, “intra_op_parallelism_threads”: 0, “allow_growth”: True } source code in https://github.com/azuredsky/rasa_nlu_gq/blob/master/rasa_nlu_gao/classifiers/embedding_intent_classifier.py