ImportError: cannot import name 'dtensor'

Hi,

Could you share how do you install the TensorFlow package?
We confirm that importing Keras with the l4t-tensorflow:r32.7.1-tf2.7-py3 container is working.

$ sudo docker run -it --rm --runtime nvidia --network host nvcr.io/nvidia/l4t-tensorflow:r32.7.1-tf2.7-py3
root@nvidia-desktop:/# python3
Python 3.6.9 (default, Dec  8 2021, 21:08:43)
[GCC 8.4.0] on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> import keras
>>> import tensorflow.keras
>>>

Thanks.