Are you using a Python Virtual environment?
If so, you’ll need to copy over a few files from the Lambda installation to your venv.
More specifically, all the libcudnn packages from /usr/lib/python3/dist-packages/tensorflow.
Just copy them over to venv_path/lib/site-packages/tensorflow
Reboot