Thanks @dakshvar22.
These important dependencies are a bit difficult to know ahead of installation. To be sure, one can do the following:
- Go to the rasa development repository.
- Open the tags.
- Find the version you want to install, say, 2.2.2.
- Download and unzip the source.
- All dependencies are in
poetry.lock
. In this case,tensorflow = ">=2.3.0,<2.4"
. - Then look up the corresponding CUDA version on TF’s installation docs, for Linux GPU, in this case.
A bit convoluted. It would help to have this more handy in the rasa documentation.
The appropriate CUDA can then be installed in a conda
environment with cudatoolkit=10.1.*
.