# Tensorflow is not detecting gpu in jetson orin

**URL:** <https://forums.developer.nvidia.com/t/tensorflow-is-not-detecting-gpu-in-jetson-orin/224070>\
**Category:** Jetson AGX Orin\
**Tags:** tensorflow, nvbugs\
**Created:** [August 16, 2022, 12:00pm UTC](https://forums.developer.nvidia.com/t/tensorflow-is-not-detecting-gpu-in-jetson-orin/224070 "2022-08-16T12:00:06Z")\
**Posts on this page:** 7\
**Page:** 1

<div class="post-metadata">

**Author:** ![soumik.layek](https://developer.download.nvidia.com/images/forums/profile-default-devtalk-84.png) [@soumik.layek](https://forums.developer.nvidia.com/u/soumik.layek)\
**Post date:** [August 16, 2022, 12:00pm UTC](https://forums.developer.nvidia.com/t/tensorflow-is-not-detecting-gpu-in-jetson-orin/224070/1 "2022-08-16T12:00:06Z")

</div>

when i run my yolov3 algorithm in tensorflow then the program is running on cpu not in gpu. Tensorflow is unable to detect any gpu in jetson orin. i have installed tensorflow as per the guidance given by the nvidia

---

<div class="post-metadata">

**Author:** ![kayccc](https://sea2.discourse-cdn.com/nvidia/user_avatar/forums.developer.nvidia.com/kayccc/32/14043_2.png) [@kayccc](https://forums.developer.nvidia.com/u/kayccc)\
**Post date:** [August 22, 2022, 12:19am UTC](https://forums.developer.nvidia.com/t/tensorflow-is-not-detecting-gpu-in-jetson-orin/224070/4 "2022-08-22T00:19:54Z")

</div>

Your topic was posted in the wrong category. I am moving this to the Jetson AGX Orin category for visibility.

---

<div class="post-metadata">

**Author:** ![AastaLLL](https://sea2.discourse-cdn.com/nvidia/user_avatar/forums.developer.nvidia.com/aastalll/32/14043_2.png) [@AastaLLL](https://forums.developer.nvidia.com/u/AastaLLL)\
**Post date:** [August 22, 2022, 3:01am UTC](https://forums.developer.nvidia.com/t/tensorflow-is-not-detecting-gpu-in-jetson-orin/224070/5 "2022-08-22T03:01:22Z")

</div>

Hi,

May I know how you install the TensorFlow package?

We tested the version `2.9.1+nv22.07` package and it can detect Orin GPU architecture correctly.

```auto
$ sudo pip3 install --pre --extra-index-url https://developer.download.nvidia.com/compute/redist/jp/v502 tensorflow==2.9.1+nv22.07
$ python3
>>> import tensorflow as tf
>>> tf.test.is_gpu_available()
...
2022-08-22 10:58:55.783605: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1532] Created device /device:GPU:0 with 22644 MB memory: -> device: 0, name: Orin, pci bus id: 0000:00:00.0, compute capability: 8.7
True

```

Thanks.

---

<div class="post-metadata">

**Author:** ![soumik.layek](https://developer.download.nvidia.com/images/forums/profile-default-devtalk-84.png) [@soumik.layek](https://forums.developer.nvidia.com/u/soumik.layek)\
**Post date:** [August 22, 2022, 4:12am UTC](https://forums.developer.nvidia.com/t/tensorflow-is-not-detecting-gpu-in-jetson-orin/224070/6 "2022-08-22T04:12:26Z")

</div>

i have installed tensorflow 2.10.0-rc0 by following the description mentioned in the nvidia website.  
After following your step of installing tensorflow 2.9 , it is now telling DNN library not found although it can now detect the Orin GPU. Can i install cuda 11.2 and cudnn 8.1 in jetson orin ?

---

<div class="post-metadata">

**Author:** ![AastaLLL](https://sea2.discourse-cdn.com/nvidia/user_avatar/forums.developer.nvidia.com/aastalll/32/14043_2.png) [@AastaLLL](https://forums.developer.nvidia.com/u/AastaLLL)\
**Post date:** [August 24, 2022, 3:35am UTC](https://forums.developer.nvidia.com/t/tensorflow-is-not-detecting-gpu-in-jetson-orin/224070/7 "2022-08-24T03:35:41Z")

</div>

Hi,

Sorry that you will need to use the CUDA and cuDNN package from JetPack.  
Currently, it is CUDA 11.4 and cuDNN 8.4 for Orin.

Could you share the script/test to reproduce the DNN library error?

For the previous error, the default command links to a CPU package from a third party.  
So please specify the version to get the correct package.  
We are checking this issue with our internal team. Will share more information with you.

Thanks.

---

<div class="post-metadata">

**Author:** ![soumik.layek](https://developer.download.nvidia.com/images/forums/profile-default-devtalk-84.png) [@soumik.layek](https://forums.developer.nvidia.com/u/soumik.layek)\
**Post date:** [September 5, 2022, 11:48am UTC](https://forums.developer.nvidia.com/t/tensorflow-is-not-detecting-gpu-in-jetson-orin/224070/9 "2022-09-05T11:48:30Z")

</div>

solved the problem. For orin i used below command

```auto
sudo pip3 install --pre --extra-index-url https://developer.download.nvidia.com/compute/redist/jp/v50 tensorflow==2.9.1+nv22.06

```

---

<div class="post-metadata">

**Author:** ![system](https://sea2.discourse-cdn.com/nvidia/user_avatar/forums.developer.nvidia.com/system/32/68080_2.png) [@system](https://forums.developer.nvidia.com/u/system)\
**Post date:** [September 28, 2022, 4:48am UTC](https://forums.developer.nvidia.com/t/tensorflow-is-not-detecting-gpu-in-jetson-orin/224070/11 "2022-09-28T04:48:24Z")

</div>

Closed on September 28, 2022, 4:48am UTC

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