Rasa core use multi CPU or GPU

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

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.

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

1 Like