Jetson Nanoとは1万円弱で買えるRaspberry Pi大きさ並みのGPU付きシングルボードコンピュータである。
で、GPUがあるということは当然機械学習用なわけで、早速ライブラリのインストールをした
環境
ビルドが必須なので先に以下のパッケージは入れておくこと
# apt install build-essential cmake g++
TensorFlowのインストール
公式がイメージを出しているのでそれを使うのが吉
依存ライブラリのインストール
# apt install libhdf5-serial-dev hdf5-tools
GPU版をインストールしたいのでtensorflow-gpuで
pip install --pre --extra-index-url https://developer.download.nvidia.com/compute/redist/jp/v42 tensorflow-gpu
結構時間かかるので注意
kerasのインストール
インストール自体は pip install keras
だけなのだが、scipyのインストールにコケる。
先に依存ライブラリのインストール
# apt install libatlas-base-dev gfortran
でcythonをインストールしておく必要がある。(すでにインストールされてあれば問題ない)
pip install -U cython