Thank you for the guide! it works for me in the following configuration:
Cuda 11.1.1
cuDNN 8.0.5
TensorFlow 2.4
Nvidia driver 461.09 (latest)
I haven't done any extensive testing but it starts to train and I can see the preview and I can see that it's going through iterations.