jeston nx 安装pytorch

安装 PyTorch 可以通过以下步骤在 Jetson NX 上完成：

1. 首先，确保 Jetson NX 已经连接到互联网。

2. 打开终端，执行以下命令来安装依赖项：

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

3. 接下来，安装 PyTorch。可以选择通过 pip 安装或者从源代码编译安装。

通过 pip 安装：

   sudo pip3 install torch torchvision torchaudio

从源代码编译安装（推荐）：

   git clone --recursive https://github.com/pytorch/pytorch
   cd pytorch
   git submodule sync
   git submodule update --init --recursive
   sudo pip3 install -r requirements.txt
   sudo python3 setup.py install

4. 安装完成后，可以验证 PyTorch 是否成功安装。在终端中执行以下命令：

   python3
   >>> import torch
   >>> print(torch.__version__)

如果没有报错，并且能够正确输出 PyTorch 的版本号，则表示安装成功。

请注意，由于 Jetson NX 是基于 ARM 架构的设备，因此需要通过编译源代码来安装 PyTorch。这可能需要一些时间和资源。如果你遇到了任何问题，请参考 PyTorch 的官方文档或在相关的开发者社区中寻求帮助。