I'm using NextBrain segmentation tool on our cluster which has NVIDIA H100 GPUs and I'm getting this error:

NVIDIA H100 with CUDA capability sm_90 is not compatible with the current PyTorch installation.
The current PyTorch install supports CUDA capabilities sm_37 sm_50 sm_60 sm_70 sm_75 sm_80 sm_86.
If you want to use the NVIDIA H100 GPU with PyTorch, please check the instructions at MailScanner has detected a possible fraud attempt from "secure-web.cisco.com" claiming to be https://pytorch.org/get-started/locally/

Should I contact our cluster managers in this regard or is there a specific workaround that can overcome this issue?

Best,

Amir