Rasa - Tensorflow support for CUDA 11 machines

I would like to train my Rasa model on A6000 GPU machine. The machine supports CUDA 11 versions which requires Tensorflow version>2.4. But Rasa is expecting tensorflow to be between 2.3 and 2.4 (uses Tensorflow 2.3.4). Is there a workaround to get it run on this machine?

1 Like

Hi, the workaround at the moment is to compile a tensorflow 2.3.4 from source that is compatible with CUDA 11. You can read more about this here (official documentation) and some tutorials like this one (untested and not endorsed by us, just an example :slight_smile: )