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 |