NVIDIA Tegra X1 with CUDA capability sm_53 is not compatible with the current PyTorch installation

Hello, I’m installing PyTorch on Jetson Nano with JetPack4.6.1 and cuda10.2. According to the official documentation, I installed the PyTorch 1.11.0a0+bfe5ad28 [Linalg] Add a runtime switch to let pytorch prefer a backend impl in… · pytorch/pytorch@bfe5ad2 · GitHub.

But when I using pytorch with cuda to run yolov3, I got the problem that

    Found GPU0 NVIDIA Tegra X1 which is of cuda capability 5.3.
    PyTorch no longer supports this GPU because it is too old.
    The minimum cuda capability supported by this library is 6.2.

NVIDIA Tegra X1 with CUDA capability sm_53 is not compatible with the current PyTorch installation.
The current PyTorch install supports CUDA capabilities sm_62 sm_72.

how should I do to solve this problem.
Thanks.

my hardware info:

Hi @user135736, can you try installing this PyTorch wheel from this post instead?

PyTorch v1.10.0

  • JetPack 4.4 (L4T R32.4.3) / JetPack 4.4.1 (L4T R32.4.4) / JetPack 4.5 (L4T R32.5.0) / JetPack 4.5.1 (L4T R32.5.1) / JetPack 4.6 (L4T R32.6.1)
2 Likes

Thanks for your answer! I had just installed Pytorch v1.10.0, and it worked successfully!

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