I have followed this guide to install CUDA on my Jetson TX2. And followed this guide to install tensorflow 2.3.1 with GPU enabled. But tensorflow doesn’t load GPU as it cannot find the CuDNN library files: root@balena:/usr/app# python3 Python 3.6.9 (default, Jan 26 2021, 15:33:00) [GCC 8.4.0] on…

Hello, if you’re using a balena base image in your Dockerfile, you should be able to add: RUN apt-get install -y libcudnn8 nvidia-cudnn8. If that does not work, please post your Dockerfile so we can take a look.

Install CuDNN on Jetson TX2

Product support balenaOS

scarlyon July 22, 2021, 8:47pm 21

@alanb128 @toochevere

I have just created another ticket with detailed explanation - Jetson: Support Nvidia Docker Images

Thanks!

Topic		Replies	Views
Problem installing NVIDIA cuDNN on Jetson Nano Product support	4	5285	June 19, 2019
Installing CUDA on Jetson base images doesn't work as instructions say it should Product support docker , jetson	5	505	February 20, 2024
Balena OS Jetson Nano Tensorflow Product support	29	3924	March 27, 2020
How to use Nvidia Jetson devices on balena Discussions	3	480	April 1, 2022
Jetson Nano GPU: CUDA driver does not match CUDA version Project help jetson	3	1883	June 1, 2022

Install CuDNN on Jetson TX2

Related topics