Some observations from the upgrade to Rasa 2.0.
Rasa 2.0 installs tensorflow 2.3.1. which has specific environment requirements before it can utilize a GPU. Requirements listed here:
For reference my working environment is:
- GPU: GeForce GTX1070 (must be an approved GPU. see CUDA GPUs | NVIDIA Developer)
- GPU driver - 456.38 (latest will probably work)
- CUDA toolkit - 10.1.243 (must be 10.1 version)
- cuDNN - 7.6.5 (must be 7.6 version)
It is possible to make TF work with other version combinations but this will require a recompile of the toolkit.