Jetson Nano Python 3.8安装PyTorch-CUDA教程(四)

jetson nano python3.8 安装pytorch(cuda)

高版本python安装GPU-pytorch,安装conda,更新CMake
  • 需特别注意直接在nano用pip安装的pytorch是CPU版本
  • 高版本python环境安装GPU pytorch,需要用CMake编译安装
  • nano自带Cmake版本太低,无法编译,需要卸载重装
  • 建议安装conda管理python环境,这里**使用python3.8环境
  • 目前有一个未解决问题: nvidia对于tx、nano等嵌入式平台发布的pytorch(cuda)版本是带有后缀的,例如pytorch-1.6.0a0,在使用torch过程中无法与深度学习中的一些库适配,导致无法跑网络。大家有解法的话,务必评论私信。
  • 1.卸载原装CMake,并安装新版
  • 先查看CMake版本,版本需要高于3.15

    cmake version
    
  • 卸载

    sudo apt remove cmake
    
  • 另外可以从cmake的安装目录下找到install_manifest.txt,txt中保存了所有cmake的附属文件的安装目录,将txt中显示的文件全部删除

    sudo rm -rf /usr/local/share/cmake*
    
  • 下载CMake,此处我选择3.18
    网址:

    最新版:https://cmake.org/download/
    老版本:https://cmake.org/files
    github:https://github.com/Kitware/CMake/releases

  • 安装
    1.打开压缩包所在文件夹,解压包

    tar -xzvf cmake-3.18.0-rc1.tar.gz
    

    2.进入文件夹

    cd cmake-3.18.0-rc1
    

    3.运行命令:

    ./configure
    

    此处可能报找不到OpenSSL的错,解决方法是编辑CMakeLists.txt文件,在 文件开头加:

    set(CMAKE_USE_OPENSSL OFF)
    

    4.运行命令:

    make
    

    5.运行命令

    sudo make install
    
  • 验证安装

    检查CMake版本

    cmake version
    

    检查编译器路径

    sudo cmake ../
    
  • 2.在编译安装之前,先设置swap分区等操作
  • 必须建立分区,否则内存满了会导致编译出错,只能删除文件夹重新编译
  • 建议开启风扇和关闭图像界面
  • 详细步骤和其他增加性能操作,详见【nano系列】文章五(建议先看
    1.新增swapfile文件大小自定义
    sudo fallocate -l 6G /var/swapfile
    

    2.配置该文件的权限

    sudo chmod 600 /var/swapfile
    

    3.建立交换分区

    sudo mkswap /var/swapfile
    

    4.启用交换分区

    sudo swapon /var/swapfile
    
  • 3.源码编译安装pytorch1.7
  • 递归下载pytorch源码及其分支

  • 可能会遇到网络问题:

  • 可以试试手机热点或者电脑搭个梯下载再传到nano

    git clone --recursive --branch 1.7 http://github.com/pytorch/pytorch
    
  • 进入文件夹,安装依赖

    python3.8 -m pip install -r requirements.txt
    
  • 最后编译安装

    python3.8 setup.py install
    
  • 经过几个小时的等待,最后显示的是:

    Installing convert-caffe2-to-onnx script to 你的环境
    Installing convert-onnx-to-caffe2 script to 你的环境
    
  • 命令行进入python环境查询cuda

    import torch
    torch.cuda.is_available()
    
  • 显示True则大功告成

  • 物联沃分享整理
    物联沃-IOTWORD物联网 » Jetson Nano Python 3.8安装PyTorch-CUDA教程(四)

    发表评论