Tesla K80 running Ubuntu 20, CUDA 11.2 Locking up on Tensorflow v2.2

I’m trying to use tensorflow v2.2 with a tesla k80, and it looks like I have everything working right, but the machine freezes up when I try to train a network.

According to nvidia-smi, and nvidia-settings all good:
±----------------------------------------------------------------------------+
| NVIDIA-SMI 460.39 Driver Version: 460.39 CUDA Version: 11.2 |
|-------------------------------±---------------------±---------------------+
| GPU Name Persistence-M| Bus-Id Disp.A | Volatile Uncorr. ECC |
| Fan Temp Perf Pwr:Usage/Cap| Memory-Usage | GPU-Util Compute M. |
| | | MIG M. |
|===============================+======================+======================|
| 0 Tesla K80 Off | 00000000:03:00.0 Off | Off |
| N/A 50C P8 28W / 149W | 135MiB / 12206MiB | 4% Default |
| | | N/A |
±------------------------------±---------------------±---------------------+
| 1 Tesla K80 Off | 00000000:04:00.0 Off | Off |
| N/A 52C P8 31W / 149W | 4MiB / 12206MiB | 0% Default |
| | | N/A |
±------------------------------±---------------------±---------------------+

±----------------------------------------------------------------------------+
| Processes: |
| GPU GI CI PID Type Process name GPU Memory |
| ID ID Usage |
|=============================================================================|
| 0 N/A N/A 842 G /usr/lib/xorg/Xorg 48MiB |
| 0 N/A N/A 1170 G /usr/bin/gnome-shell 84MiB |
| 1 N/A N/A 842 G /usr/lib/xorg/Xorg 3MiB |
±----------------------------------------------------------------------------+

What might be the issue?

Hi,

With this web page you can realize that CUDA 11 requires TF >= 2.4.0:

Regards