在安装TensorFlow GPU之前,需要先安装一些依赖包。
-
安装pip
因为Jetson Nano中已经安装了Python3.6版本,所以安装pip还是比较简单的sudo apt-get install python3-pip python3-dev python3 -m pip install --upgrade pip
安装后pip后,需要把它升级到最新版,升级后pip版本为19.3.1。这里面升级后会有一个小Bug,需要手动改一下,升级后都得改,其他版本的也是,通病了。
sudo vim /usr/bin/pip3
将原来的
from pip import main if __name__ == '__main__':sys.exit(main())
改成
from pip import __main__if __name__ == '__main__':sys.exit(__main__._main())
修改结束后保存。运行pip3 -V成功后显示
-
安装依赖包
sudo apt-get install libhdf5-serial-dev hdf5-tools
-
安装TensorFlow
我们现在英伟达官网上下载包,不然安装要老半天了,下面是链接,建议用迅雷,只要速度快,用什么都行。
tf_gpu-1.14.0+nv19.10-py3
然后执行命令安装,至于加个清华源是为了防止缺某些依赖包,需要下载安装,如果用官方源的话慢,用上清华源可以快很多。sudo pip3 install -i https://pypi.tuna.tsinghua.edu.cn/simple tensorflow_gpu-1.14.0+nv19.10-cp36-cp36m-linux_aarch64.whl
安装完成后,随便找个代码试试,没错就Ok了。
-
常用包安装下载
建议一个一个分开来安装,不然容易卡死,我为了方便打,就放在一起了。sudo pip3 install -i https://pypi.tuna.tsinghua.edu.cn/simple Cython numpy scipy scikit-image matplotlib ipython h5py leveldb networkx nose pandas python-dateutil protobuf python-gflags pyyaml Pillow six