Xavier NX において、機械学習には欠かせない TensorFlow による GPU を利用した、Python プログラムの実行環境構築について記述します。下記のブログを参考にさせていただきました。
【jetson XAVIER NX】tensorflowでGPUを利用できる環境を構築
| 1 | sudo apt-cache show nvidia-jetpack | 4.5 |
| 2 | python3 -V | 3.6.9 |
| 3 | pip3 -V | 21.2.3 |
| sudo -H pip3 install virtualenv | if you need : use -H | |
| virtualenv tf_env | if exist tf\env : rm -R tf_env | |
| source tf_env/bin/activate |
| 12 | python3 |
| 13 | >>>import tensorflow as tf >>> tf.__version__ |
| 14 | >>>from tensorflow import keras >>> keras.__version__ |
| 15 | >>>from tensorflow.python.client import device_lib >>>device_lib.list_local_devices() |
Official TensorFlow for Jetson AGX XavierNX
| 6 | sudo apt-get update | |
| 7 | sudo apt-get install libhdf5-serial-dev hdf5-tools libhdf5-dev zlib1g-dev zip libjpeg8-dev liblapack-dev libblas-dev gfortran | |
| 8 | sudo apt-get install python3-pip | |
| 9 | sudo pip3 install -U pip testresources setuptools==49.6.0 | warning |
| 10 | sudo pip3 install -U numpy==1.16.1 future==0.18.2 mock==3.0.5 h5py==2.10.0 keras_preprocessing==1.1.1 keras_applications==1.0.8 gast==0.2.2 futures protobuf pybind11 | warning |
| 11 | sudo pip3 install –pre –extra-index-url https://developer.download.nvidia.com/compute/redist/jp/v45 tensorflow |
venv: Python 仮想環境管理
| 4 | python3 -m venv tf_env | |
| 5 | source tf_env/bin/activate | |
| pip install [package name] | ||
| pip freeze | ||
| deactivate |