Jetson Nano特定配置下是否有PyTorch预编译wheel包？

你的配置（JetPack 4.6.6、CUDA 10.2、Python 3.6.9、Ubuntu 18.04）有现成的预编译PyTorch wheel包，完全不需要耗时的源码编译，具体方案如下：

推荐版本与安装步骤 #

• 适配该环境的稳定版本为PyTorch 1.8.0，配套的torchvision、torchaudio版本分别为0.9.0和0.8.0，这些包都是针对Jetson平台预编译并内置CUDA 10.2支持的。
• 先安装依赖库：

  sudo apt-get update
  sudo apt-get install libopenblas-base libopenmpi-dev -y
  

• 执行安装命令（注意不要加CPU源参数）：

  pip3 install torch==1.8.0 torchvision==0.9.0 torchaudio==0.8.0
  

验证安装有效性 #

安装完成后，运行以下代码确认CUDA支持是否正常：

import torch
print(torch.cuda.is_available())
print(torch.version.cuda)

若输出True和10.2，则说明带CUDA加速的PyTorch已成功安装。

关键说明 #

• 你之前使用--index-url https://download.pytorch.org/whl/cpu会强制拉取纯CPU版本，这是导致无CUDA支持的原因，去掉该参数即可获取适配Jetson的GPU版本。
• Jetson平台的预编译PyTorch包是针对ARM架构和JetPack/CUDA版本优化的，不仅安装速度快，运行稳定性也远高于自行编译的版本。

内容的提问来源于stack exchange，提问作者Dribbler