PyTorch support for NVIDIA RTX 5090

vision
1

Is PyTorch nightly build supporting NVIDIA RTX5090? My vision program was running fine till last week with nightly build & then it started giving errors.

I pulled stable release and it gives error below -

NVIDIA RTX 5090 with CUDA capability sm_120 is not compatible with the current PyTorch installation.
The current PyTorch install supports CUDA capabilities sm_50 sm_60 sm_70 sm_75 sm_80 sm_86 sm_90.
If you want to use the NVIDIA GeForce RTX 5090 GPU with PyTorch, please check the instructions at https://pytorch.org/get-started/locally/

warnings.warn(…

2

Yes, all of our nightly and stable releases using CUDA 12.8+ support Blackwell architectures.
Based on the error message it seems you are using a PyTorch binary with CUDA <=12.6.

1 Like