Tensorflow 2.5 will require CUDA 11.2, which isn't supported by Google's deep learning image stack yet.
Hence, downgrade it to tensorflow 2.4.2 which will work with CUDA 11.0.
Unfortunately we don't have automated testing configured for the code in this
repository yet so we set up this checklist as an automatic reminder:
[ ] Ensure that the smoke tests pass using the current (or upcoming) CDR
[ ] Update documentation relevant to this pull request
Questions? See CONTRIBUTING.md
or file an issue so that we can get it documented!
https://broadworkbench.atlassian.net/browse/IA-2686
Tensorflow 2.5 will require CUDA 11.2, which isn't supported by Google's deep learning image stack yet. Hence, downgrade it to tensorflow 2.4.2 which will work with CUDA 11.0.
Unfortunately we don't have automated testing configured for the code in this repository yet so we set up this checklist as an automatic reminder:
Questions? See CONTRIBUTING.md or file an issue so that we can get it documented!