jetson 安装pytorch

要在Jetson设备上安装PyTorch，可以按照以下步骤进行操作：

1. 首先，确保你的Jetson设备已经安装了合适版本的JetPack。PyTorch通常与特定版本的JetPack捆绑在一起。你可以在NVIDIA官方网站上找到与你的Jetson设备兼容的JetPack版本。

2. 接下来，打开终端并使用以下命令安装依赖项：

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

3. 然后，使用pip安装PyTorch。根据你的Jetson设备型号，选择相应的命令：
- 对于Jetson Nano：

   sudo pip3 install torch torchvision torchaudio

- 对于Jetson TX1/TX2：

   sudo pip3 install torch==1.8.0 torchvision==0.9.0 torchaudio==0.8.0 -f https://developer.download.nvidia.com/compute/redist/jp/v44

- 对于Jetson Xavier NX/AGX Xavier：

   sudo pip3 install torch torchvision torchaudio

4. 安装完成后，你可以通过导入PyTorch模块来验证安装是否成功：

python3
import torch
print(torch.__version__)

如果没有报错，并且能够输出PyTorch的版本号，则说明PyTorch已成功安装在你的Jetson设备上。

希望这对你有帮助！如果有任何进一步的问题，请随时提问。