DGX Dashboard Playbook: pytorch in sample code not supporting cuda 12.1?

Accelerated Computing DGX Spark / GB10 User Forum DGX Spark / GB10
1

When following the DGX Dashboard Playbook: DGX Dashboard | DGX Spark

After Juypter launches and creating a test notebook and pasting the provided sample code in cell and running with Shift + Enter, it appears to show a user warning:

UserWarning: 
    Found GPU0 NVIDIA GB10 which is of cuda capability 12.1.
    Minimum and Maximum cuda capability supported by this version of PyTorch is
    (8.0) - (12.0)

I assume this sample code is not using the version of Pytorch that supports 12.1 and needs to be updated? Is there a workaround?

2

You can safely ignore. cuda 12.1 is newer than the pytorch version used here. Eng will be fixing this in a future OTA.

1 Like
3

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