Jetpack 5.0.2 Tensorflow is_built_with_cuda = False

Hello,

I installed jetpack with the sdkmanager to version 5.0.2 after installation I follow directly the Installation Tutorial from Installing TensorFlow for Jetson Platform :: NVIDIA Deep Learning Frameworks Documentation. I change the download link to v52 and download tensorflow 2.11.0rc2 (tried other versions as well …). I Use a NVIDIA Jetson AGX Xavier. Installing is not problem but when I import tensorflow it wont use CUDA to build with GPU. Is this a known problem? Is there any way to solve it? Please help.

Hi,

This is a known issue.
The RC2 package is not our prebuilt but a CPU version released by third-party.

Please update the command with a specific version to get our prebuilt:
For example:

$ sudo pip3 install --pre --extra-index-url https://developer.download.nvidia.com/compute/redist/jp/v502 tensorflow==$TF_VERSION+nv$NV_VERSION
$ sudo pip3 install --pre --extra-index-url https://developer.download.nvidia.com/compute/redist/jp/v502 tensorflow==2.10.0+nv22.10

Thanks.

1 Like