2018年4月25日 星期三

nVIDIA Jetson TX2 jetpack3.2 tensorflow tensorboard 安裝與使用

nVIDIA Jetson TX2 推出的早期,tensorflow 安裝須要自己 build,非常的花時間。

現在 jetpack 版本到 3.2,nVIDIA官方很貼心的提供了 wheel 檔來讓社群的朋友們使用。

釋出的位置如下
https://devtalk.nvidia.com/default/topic/1031300/jetson-tx2/tensorflow-1-7-wheel-with-jetpack-3-2-/

下載位置如下
https://nvidia.app.box.com/v/TF170-py35-wTRT



下載後得到 tensorflow-1.7.0-cp35-cp35m-linux_aarch64.whl 這個檔案

接著使用下面指令安裝
pip3 install tensorflow-1.7.0-cp35-cp35m-linux_aarch64.whl

安裝好後 使用 pip3 list | grep tensor 就可以看到 tensorflow 和 tensorboard了

之後進入 python3 並使用下面程式碼就可以看到 tensorflow 的版本了
import tensorflow as tf;
print(tf.__version__)


接下來輪到 tensorboard 了

上面我們用 pip3 list 去查的時候,已經確認 tensorboard 有安裝,不過實際用的時候,下bash 指令
tensorboard
系統回答 bash: tensorboard: command not found

這時候就要自己弄了
步驟如下:

  1. 1.pip3 show tensorboard 尋找tensorboard 所存放的location

    將會得到
    Location: /home/nvidia/.local/lib/python3.5/site-packages
  2. ls /home/nvidia/.local/lib/python3.5/site-packages/tensorboard
    我們在這邊可以找到main.py
  3. 用 alias 加到~/.bashrc裡面,也就是在.bashrc最下面加入一行

    alias tensorboard='python3 /home/nvidia/.local/lib/python3.5/site-packages/tensorboard/main.py'


    之後再到 bash下指令
    source ~/.bashrc
    重新載入 .bashrc的東西

    之後就可以在 bash command line 下 tensorboard 的指令了

沒有留言:

張貼留言