jetson orin nx怎么安装gpu版本pytorch

安装GPU版本的PyTorch需要以下步骤：

1. 确保你的Jetson Orin NX上安装了CUDA和cuDNN。你可以通过运行以下命令来检查它们的安装情况：

   nvcc --version

   cat /usr/local/cuda/version.txt

2. 安装Python3和pip3（如果没有安装的话）：

   sudo apt-get update
   sudo apt-get install python3-dev python3-pip

3. 安装PyTorch：

   pip3 install torch torchvision torchaudio -f https://download.pytorch.org/whl/cu111/torch_stable.html

这将安装与CUDA 11.1兼容的PyTorch版本。如果你的Jetson Orin NX上安装了不同版本的CUDA，请相应地更改上面的URL。

4. 检查是否成功安装了GPU版本的PyTorch：

   python3 -c "import torch; print(torch.cuda.is_available())"

如果输出为True，则说明你已经成功安装了GPU版本的PyTorch。

希望这能帮助你安装所需的PyTorch版本。