Cannot import TF 2.6.0 correctly on Xavier NX

hi trying to install tensor 2.6.0 with

pip3 install --pre --extra-index-url tensorflow==2.6.0+nv21.11

but getting the following error:

Using cached h5py-3.1.0.tar.gz (371 kB)
Installing build dependencies … error
ERROR: Command errored out with exit status 1:
command: /usr/bin/python3 /usr/local/lib/python3.6/dist-packages/pip install --ignore-installed --no-user --prefix /tmp/pip-build-env-k_tjewda/overlay --no-warn-script-location --no-binary :none: --only-binary :none: -i Simple index --extra-index-url --pre – setuptools wheel
cwd: None
Complete output (31 lines):
Traceback (most recent call last):
File “/usr/lib/python3.6/”, line 193, in _run_module_as_main
“main”, mod_spec)
File “/usr/lib/python3.6/”, line 85, in _run_code
exec(code, run_globals)
File “/usr/local/lib/python3.6/dist-packages/pip/”, line 29, in
from pip._internal.cli.main import main as _main
File “/usr/local/lib/python3.6/dist-packages/pip/_internal/cli/”, line 9, in
from pip._internal.cli.autocompletion import autocomplete
File “/usr/local/lib/python3.6/dist-packages/pip/_internal/cli/”, line 10, in
from pip._internal.cli.main_parser import create_main_parser
File “/usr/local/lib/python3.6/dist-packages/pip/_internal/cli/”, line 8, in
from pip._internal.cli import cmdoptions
File “/usr/local/lib/python3.6/dist-packages/pip/_internal/cli/”, line 23, in
from pip._internal.cli.parser import ConfigOptionParser
File “/usr/local/lib/python3.6/dist-packages/pip/_internal/cli/”, line 12, in
from pip._internal.configuration import Configuration, ConfigurationError
File “/usr/local/lib/python3.6/dist-packages/pip/_internal/”, line 26, in
from pip._internal.utils.logging import getLogger
File “/usr/local/lib/python3.6/dist-packages/pip/_internal/utils/”, line 13, in
from pip._internal.utils.misc import ensure_dir
File “/usr/local/lib/python3.6/dist-packages/pip/_internal/utils/”, line 36, in
from pip._vendor.tenacity import retry, stop_after_delay, wait_fixed
File “/usr/local/lib/python3.6/dist-packages/pip/_vendor/tenacity/”, line 26, in
from concurrent import futures
File “/usr/local/lib/python3.6/dist-packages/concurrent/futures/”, line 8, in
from concurrent.futures._base import (FIRST_COMPLETED,
File “/usr/local/lib/python3.6/dist-packages/concurrent/futures/”, line 357
raise type(self._exception), self._exception, self._traceback
SyntaxError: invalid syntax
WARNING: Discarding (from Links for h5py) (requires-python:>=3.6). Command errored out with exit status 1: /usr/bin/python3 /usr/local/lib/python3.6/dist-packages/pip install --ignore-installed --no-user --prefix /tmp/pip-build-env-k_tjewda/overlay --no-warn-script-location --no-binary :none: --only-binary :none: -i Simple index --extra-index-url --pre – setuptools wheel Check the logs for full command output.
ERROR: Could not find a version that satisfies the requirement h5py~=3.1.0 (from tensorflow) (from versions: 2.2.1, 2.3.0b1, 2.3.0, 2.3.1, 2.4.0b1, 2.4.0, 2.5.0, 2.6.0, 2.7.0rc2, 2.7.0, 2.7.1, 2.8.0rc1, 2.8.0, 2.9.0rc1, 2.9.0, 2.10.0, 3.0.0rc1, 3.0.0, 3.1.0)
ERROR: No matching distribution found for h5py~=3.1.0