How to install tensorflow with GPU support on Jetson Orin Nano?

Autonomous Machines Jetson & Embedded Systems Jetson Orin Nano
1

I have installed tensorflow as below, but the code runs on CPU not GPU. How can I fix it?

BTW, I DID check Installing TensorFlow for Jetson Platform - NVIDIA Docs , but there is no commands for Jetson Orin Nano board.

$ pip install tensorflow

$ python
Python 3.10.12 (main, Nov 20 2023, 15:14:05) [GCC 11.4.0] on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> import tensorflow as tf
>>> tf.__version__
'2.16.1'
>>> tf.__path__
['/home/daniel/.local/lib/python3.10/site-packages/keras/api/_v2', '/home/daniel/.local/lib/python3.10/site-packages/keras/_tf_keras', '/home/daniel/.local/lib/python3.10/site-packages/tensorflow', '/home/daniel/.local/lib/python3.10/site-packages/tensorflow/_api/v2']
>>>

图片
图片1207×456 132 KB

EDIT1: TF 2.16.1 Fails to work with GPUs · Issue #63362 · tensorflow/tensorflow · GitHub
EDIT2: Or there is no GPU version for tensorflow 2.16.1 - python 3.10??? Buat dari sumber  |  TensorFlow

3

Hi,

Please follow the installation steps on the document.
For JetPack 6, the command should be:

$ sudo pip3 install --extra-index-url https://developer.download.nvidia.com/compute/redist/jp/v60dp tensorflow==2.15.0+nv24.03

Thanks.

5

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