thor jetpack7的torch gpu配置

Question

您好，想和您请教一下有关thor jetpack7的torch gpu配置的问题
1.想自己配置gpu环境但是找不到官方/非官方发布的适配jp7的r38的docker
2.想通过wheel包安装torch-gpu，会遇到torch和jp7版本不适配导致只能使用cpu-only而没办法使用gpu的现象

Answer 1

针对 Thor（Jetson Thor）搭配 JetPack 7（JP7）的 PyTorch GPU 环境配置问题，解决方案如下：
问题 1：适配 JP7 的 r38 Docker 镜像缺失
JetPack 7 对应 L4T R38 版本，目前官方 / 社区针对 Jetson Thor 的 Docker 镜像资源较少，可通过以下方式解决：
基于官方基础镜像构建：
以 NVIDIA 提供的nvcr.io/nvidia/l4t-base:r38.0.0为基础镜像，手动安装 CUDA、cuDNN（JP7 自带的版本）及 PyTorch 依赖，自行构建 Docker 镜像。
社区资源跟进：
关注 NVIDIA NGC、JetsonHacks 等社区，JP7 发布后相关 Docker 镜像通常会逐步更新。
问题 2：PyTorch wheel 包与 JP7 版本不兼容
JetPack 7 的 L4T R38 对应特定的 PyTorch 版本，需选择适配的 wheel 包：