No module named 'torch._C._distributed_c10d'; 'torch._C' is not a package

Robotics & Edge Computing Jetson & Embedded Systems Jetson AGX Orin
1
  1. 安装NVIDIA提供的PyTorch版本2.1.0，运行stable Diffusion， 会报错No module named ‘torch._C._distributed_c10d’; ‘torch._C’ is not a package。
  2. 尝试安装不是NVIDIA提供的PyTorch 2.1.0版本，不会报错，但torch.cuda.is_available()显示false，无法使用GPU。
  3. 请问 No module named ‘torch._C._distributed_c10d’; ‘torch._C’ is not a package。这个error怎么解决。
  4. 环境：Python3.8, torchvision 0.16, cuda11.4, 硬件：Jetson Orin Developer Kits
4

Hi,

Could you upgrade your environment to JetPack 6 and test a newer PyTorch prebuilt to see if the same issue exists?

Thanks.

7

There is no issue when upgrade environment to JetPack 6, thanks!

9

This topic was automatically closed 14 days after the last reply. New replies are no longer allowed.