当前位置: 代码迷 >> 综合 >> 【剁手系列】Jetson Nano(二)安装TensorFlow GPU以及一些常用包
  详细解决方案

【剁手系列】Jetson Nano(二)安装TensorFlow GPU以及一些常用包

热度:34   发布时间:2023-09-05 19:22:21.0

在安装TensorFlow GPU之前,需要先安装一些依赖包。

  1. 安装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成功后显示
    【剁手系列】Jetson Nano(二)安装TensorFlow GPU以及一些常用包

  2. 安装依赖包

    sudo apt-get install libhdf5-serial-dev hdf5-tools
    
  3. 安装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了。

  4. 常用包安装下载
    建议一个一个分开来安装,不然容易卡死,我为了方便打,就放在一起了。

    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
    
  相关解决方案