NVIDIA Jetson JetPack 5.0.1に TensorFlowを NVIDIA公式の手順でインストールする方法

【2020年版】NVIDIA Jetson Nano、Jetson Xavier NXの便利スクリプト

# JetPack 5.0.2

$ uname -a
Linux ubuntu 5.10.104-tegra #1 SMP PREEMPT Wed Aug 10 20:17:07 PDT 2022 aarch64 aarch64 aarch64 GNU/Linux

$ /usr/local/cuda/bin/nvcc -V
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2022 NVIDIA Corporation
Built on Wed_May__4_00:02:26_PDT_2022
Cuda compilation tools, release 11.4, V11.4.239
Build cuda_11.4.r11.4/compiler.31294910_0

# Install system packages required by TensorFlow
sudo apt-get -y update

sudo apt-get -y install libhdf5-serial-dev hdf5-tools libhdf5-dev zlib1g-dev zip libjpeg8-dev liblapack-dev libblas-dev gfortran

# Install and upgrade pip3
sudo apt-get -y install python3-pip
sudo pip3 install -U pip testresources setuptools==65.5.0
# Successfully installed pbr-5.11.0 pip-22.3.1 setuptools-65.5.0 testresources-2.0.1

# Installing TensorFlow
# TensorFlow 2.x package version 2.10.0

# Install the Python package dependencies
# TensorFlow 2.x package version 2.10.0
sudo pip3 install -U numpy==1.21.1 future==0.18.2 mock==3.0.5 keras_preprocessing==1.1.2 keras_applications==1.0.8 gast==0.4.0 protobuf pybind11 cython pkgconfig packaging h5py==3.6.0

# Installing TensorFlow
# TensorFlow 2.x package version 2.10.0
sudo pip3 install --extra-index-url https://developer.download.nvidia.com/compute/redist/jp/v502 tensorflow==2.10.0+nv22.11

# Installing TensorFlow
# TensorFlow 1.x package version 1.15.5
# ERROR: pip's dependency resolver does not currently take into account all the packages that are installed. This behaviour is the source of the following dependency conflicts.
# tensorflow 1.15.5+nv22.11 requires gast==0.3.3, but you have gast 0.4.0 which is incompatible.
# tensorflow 1.15.5+nv22.11 requires h5py==2.10.0, but you have h5py 3.6.0 which is incompatible.
sudo pip3 install -U numpy==1.21.1 future==0.18.2 mock==3.0.5 keras_preprocessing==1.1.2 keras_applications==1.0.8 gast==0.3.3 protobuf==3.19.6 pybind11 cython pkgconfig packaging h5py==2.10.0
# Successfully installed cython-0.29.32 gast-0.3.3 h5py-2.10.0 keras_applications-1.0.8 keras_preprocessing-1.1.2 mock-3.0.5 numpy-1.21.1 packaging-21.3 pkgconfig-1.5.5 protobuf-3.19.6 pybind11-2.10.1

# Installing TensorFlow
# TensorFlow 1.x package version 1.15.5
sudo pip3 install --extra-index-url https://developer.download.nvidia.com/compute/redist/jp/v502 'tensorflow==1.15.5+nv22.11'
# Successfully installed MarkupSafe-2.1.1 absl-py-1.3.0 astor-0.8.1 astunparse-1.6.3 gast-0.3.3 google-pasta-0.2.0 grpcio-1.51.1 h5py-2.10.0 importlib-metadata-5.1.0 markdown-3.4.1 opt-einsum-3.3.0 tensorboard-1.15.0 tensorflow-1.15.5+nv22.11 tensorflow-estimator-1.15.1 termcolor-2.1.1 werkzeug-2.2.2 wrapt-1.14.1 zipp-3.11.0

# Check TensorFlow

$ python3 -c "import tensorflow"
2022-12-07 08:34:35.613751: I tensorflow/stream_executor/platform/default/dso_loader.cc:49] Successfully opened dynamic library libcudart.so.11.0
WARNING:tensorflow:Deprecation warnings have been disabled. Set TF_ENABLE_DEPRECATION_WARNINGS=1 to re-enable them.

# Uninstalling TensorFlow
sudo pip3 uninstall -y tensorflow

NVIDIA Jetsonで matching-with-gansを動かす方法

NVIDIA Jetson JetPack 5.0.1に PyTorchを NVIDIA公式の手順でインストールする方法

NVIDIA Jetsonで音声認識の Whisperを動かす方法

NVIDIA Jetsonで流行の AIお絵かきの min(DALL・E)を動かす方法