Revision 6cc140ece3d5c153ad7baaf4c95c9c9f295d01ad authored by Henning Becker on 02 April 2024, 16:38:14 UTC, committed by Copybara-Service on 02 April 2024, 16:39:00 UTC
CudnnRNN and CudnnRNNV2 are not compatible with cuDNN 9+, so
this change makes Sonnet use CudnnRNNV3 instead.

Note that this raises the minimum supported cuDNN version to 8.1
(which is below 8.9 - the minimum supported cuDNN version in Tensorflow anyway).

PiperOrigin-RevId: 621206198
1 parent 26b0518
History
File Mode Size
.github
docs
examples
sonnet
CONTRIBUTING.md -rw-r--r-- 1.7 KB
LICENSE -rw-r--r-- 11.1 KB
MANIFEST.in -rw-r--r-- 34 bytes
README.md -rw-r--r-- 11.4 KB
WORKSPACE -rw-r--r-- 27 bytes
readthedocs.yml -rw-r--r-- 357 bytes
requirements-test.txt -rw-r--r-- 46 bytes
requirements-tf.txt -rw-r--r-- 53 bytes
requirements.txt -rw-r--r-- 74 bytes
setup.py -rw-r--r-- 2.0 KB
test.sh -rwxr-xr-x 1.8 KB

README.md

back to top