Jetson Orin Nano can not detect GPU with tensorflow

Hi, I am using Jetson Orin Nano with Jetpack version 6.2 and cuda-toolkit 12.6. I installed tensorflow to my python venv by using codes :

$ sudo apt-get update
$ sudo apt-get install libhdf5-serial-dev hdf5-tools libhdf5-dev zlib1g-dev zip libjpeg8-dev liblapack-dev libblas-dev gfortran
$ sudo apt-get install python3-pip
$ sudo python3 -m pip install --upgrade pip
$ sudo pip3 install -U testresources setuptools==65.5.0
$ pip3 install -U numpy grpcio absl-py py-cpuinfo psutil portpicker six mock requests gast h5py astor termcolor protobuf keras-applications keras-preprocessing wrapt google-pasta setuptools testresources
$ pip3 install --extra-index-url https://developer.download.nvidia.com/compute/redist/jp/v62 tensorflow

Also I tried :

$ pip3 install --extra-index-url https://developer.download.nvidia.com/compute/redist/jp/v62 tensorflow==2.17.0+nv25.02 (When I use this command I got an error :

ERROR: Could not find a version that satisfies the requirement tensorflow==2.17.0+nv25.02 (from versions: 2.10.0rc0, 2.10.0rc1, 2.10.0rc2, 2.10.0rc3, 2.10.0, 2.10.1, 2.11.0rc0, 2.11.0rc1, 2.11.0rc2, 2.11.0, 2.11.1, 2.12.0rc0, 2.12.0rc1, 2.12.0, 2.12.1, 2.13.0rc0, 2.13.0rc1, 2.13.0rc2, 2.13.0, 2.13.1, 2.14.0rc0, 2.14.0rc1, 2.14.0, 2.14.1, 2.15.0rc0, 2.15.0rc1, 2.15.0, 2.15.1, 2.16.0rc0, 2.16.1, 2.16.2, 2.17.0rc0, 2.17.0rc1, 2.17.0, 2.17.1, 2.18.0rc0, 2.18.0rc1, 2.18.0rc2, 2.18.0, 2.18.1, 2.19.0rc0, 2.19.0, 2.20.0rc0) 12.6.68-1 arm64 CUDA nvcc
ERROR: No matching distribution found for tensorflow==2.17.0+nv25.02 arm64 CUDA disassemb
)

But despite processing these steps, command : print(tf.config.list_physical_devices()) still returns,

[PhysicalDevice(name=‘/physical_device:CPU:0’, device_type=‘CPU’)]

what should I do ?

Hi,

Please update the extra-index-url to https://developer.download.nvidia.com/compute/redist/jp/v61 and try it again.

Thanks.

Using https://developer.download.nvidia.com/compute/redist/jp/v61/tensorflow/tensorflow-2.16.1+nv24.08-cp310-cp310-linux_aarch64.whl both GPU & CPU are listed as available:

[PhysicalDevice(name=‘/physical_device:CPU:0’, device_type=‘CPU’), PhysicalDevice(name=‘/physical_device:GPU:0’, device_type=‘GPU’)]

Using tf-2.20 only the CPU shows when listing devices.

1 Like

Hi,

We only have v2.16 prebuilt package:
https://developer.download.nvidia.com/compute/redist/jp/v61/tensorflow/

Thanks.

This topic was automatically closed 14 days after the last reply. New replies are no longer allowed.