[Feature Request] RTX 50 Series (Blackwell/sm_120) GPU Support

[BoletuPeng]

Hi,

I'm trying to run DeepPrep on an RTX 5090 (Blackwell architecture, sm_120) but encountered CUDA compatibility issues since the current Docker image uses PyTorch that only supports up to sm_90.

I've created a guide with workarounds to get DeepPrep running on Blackwell GPUs:
https://github.com/BoletuPeng/DeepPrep-blackwell-support

Would it be possible to:

1. Update the official Docker image to support CUDA 12.8+ / PyTorch 2.7+?
2. Or add documentation for RTX 50 series users?

Happy to help if needed.

[lincong8722]

Hi,
We are pleased to inform you that DeepPrep now supports execution on NVIDIA 50-series graphics cards. Please pull the Docker image version 25.1.1.cuda129 to proceed with your testing.
Best regards

[BoletuPeng]

Excuse me. Are you saying that you had already completed the 50-series support before I spent time doing the migration, but just didn’t tell anyone? Or did you ‘coincidentally’ finish the migration the day after I made my post?