動かざることバグの如し

近づきたいよ 君の理想に

Jetson NanoにTensorFlowとKerasをインストール

Jetson Nanoとは1万円弱で買えるRaspberry Pi大きさ並みのGPU付きシングルボードコンピュータである。

で、GPUがあるということは当然機械学習用なわけで、早速ライブラリのインストールをした

環境

  • Ubuntu 18.04
  • CUDA 10 最初から入ってる
  • python 3.6 pyenv経由でインストールした

ビルドが必須なので先に以下のパッケージは入れておくこと

# 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