Nvidia jetson-Xavier 安装torch

一、配置刷机

二、安装torch、torchvison

详情参考以下官方链接：
​PyTorch for Jetson - version 1.10 now available - Jetson Nano - NVIDIA Developer ForumsBelow are pre-built PyTorch pip wheel installers for Python on Jetson Nano, Jetson TX1/TX2, and Jetson Xavier NX/AGX with JetPack 4.2 and newer. Download one of the PyTorch binaries from below for your version of JetPa…https://forums.developer.nvidia.com/t/pytorch-for-jetson-version-1-10-now-available/72048

https://download.pytorch.org/whl/torch_stable.html （这是编译好的离线安装包）

本人亲测如下：

1、下载最新版本：

2、执行安装指令：

sudo apt-get install python3-pip libopenblas-base libopenmpi-dev 
pip3 install Cython
pip3 install numpy torch-1.10.0-cp36-cp36m-linux_aarch64.whl

3、导入torch模块 测试cuda正常 

 

4、确认torch安装正确之后，接着开始进行torchvision的安装，链接如下：​GitHub - pytorch/vision at release/0.11

安装指令如下，时间可能会有点长 大概十几分钟 做好心理准备。

$ sudo apt-get install libjpeg-dev zlib1g-dev libpython3-dev libavcodec-dev libavformat-dev libswscale-dev
$ git clone --branch <version> https://github.com/pytorch/vision torchvision   # see below for version of torchvision to download #可以直接从github上下载
$ cd torchvision
$ export BUILD_VERSION=0.x.0  # where 0.x.0 is the torchvision version  这我用的版本torch 1.10 对应的vision应该是0.11.1
$ python3 setup.py install --user

ok 这一切装完之后就可以开始安装 稀疏卷积库 spconv了

首先，确认一下cmake版本：

cmake --version

 如果提示未安装，则进行安装，版本不低于 3.13.2，

下载链接：

Download | CMake

安装教程参考链接：

Ubuntu中安装Cmake - 燕清扬 - 博客园