【エラー多発】Jetson Nano 2GBにTensorFlowをインストールできるまで

Jetson Nano 2GBのセットアップが終わったのでTensorFlowをインストールすることにしました。しかし、エラーの連続でインストールに苦戦したので本記事にまとめておきたいと思います。誰かの参考になれば幸いです。

セットアップまではこちら

Contents

インストールする前に

TensorFlowをインストールする前に使用CPU数とクロック周波数を最大に設定しました。

	$ sudo nvpmodel -m 0
	$ sudo jetson_clocks

view raw jetson_clock hosted with ❤ by GitHub

スワップ領域も拡大します。

	$ git clone https://github.com/JetsonHacksNano/installSwapfile
	$ cd installSwapfile
	$ ./installSwapfile.sh

view raw swap hosted with ❤ by GitHub

スワップとはメモリが足りないときにメモリの中身をハードディスクに移し、実際よりも大きなメモリがあるようにみせることです。

はじめ、こちらのツールを使用させていただきインストールを試みました。

エラーでインストールできず。ログを確認するとh5pyが原因になっている感じでした。

色々調べていると、こちらの記事を発見。

以下のコードでインストールが上手く行きました。

	$ apt-get update
	$ apt-get install -y python3-pip pkg-config
	$ apt-get install -y libhdf5-serial-dev hdf5-tools libhdf5-dev zlib1g-dev zip libjpeg8-dev liblapack-dev libblas-dev gfortran
	$ ln -s /usr/include/locale.h /usr/include/xlocale.h
	$ pip3 install --verbose 'protobuf<4' 'Cython<3'
	$ wget --no-check-certificate https://developer.download.nvidia.com/compute/redist/jp/v461/tensorflow/tensorflow-2.7.0+nv22.1-cp36-cp36m-linux_aarch64.whl
	$ pip3 install --verbose tensorflow-2.7.0+nv22.1-cp36-cp36m-linux_aarch64.whl

view raw TensorFlow_install hosted with ❤ by GitHub

JETPACKが4.5だったのでTENSORFLOWのバージョンはここから選びました。
https://developer.download.nvidia.com/compute/redist/jp/v45/tensorflow

TensorFlowがきちんと動くかは、MNISTを使って確認をしました。

	import tensorflow as tf
	from tensorflow.keras import layers, models, datasets

	# データセットの読み込み
	(x_train, y_train), (x_test, y_test) = datasets.mnist.load_data()
	x_train, x_test = x_train / 255.0, x_test / 255.0

	# modelを準備する
	model = models.Sequential([
	layers.Flatten(input_shape=(28, 28)),
	layers.Dense(128, activation='relu'),
	layers.Dropout(0.2),
	layers.Dense(10, activation=”softmax”)
	])

	# modelをコンパイルする
	model.compile(optimizer='adam', loss=”sparse_categorical_crossentropy”, metrics=['accuracy'])

	# modelの学習
	model.fit(x_train, y_train, epochs=5)

	# 学習結果確認
	model.evaluate(x_test, y_test, verbose=1)