Jetson Nanoにtensorflow-gpuのインストール

docs.nvidia.com

sudo pip3 install --pre --extra-index-url https://developer.download.nvidia.com/compute/redist/jp/v46 tensorflow

インストールできたが、import tensorflowすると落ちる。
環境変数にOPENBLAS_CORETYPE=ARMV8を指定する。
stackoverflow.com
KerasがGPUメモリを掴んで離さない問題もあるので、TF_FORCE_GPU_ALLOW_GROWTH=true
もあわせて.bashrcに追記した。

export OPENBLAS_CORETYPE=ARMV8
export TF_FORCE_GPU_ALLOW_GROWTH=true