Sionna on Blackwell (RTX 6000 Pro): setup guidance?

[hssam1002]

Hi all,
Our lab recently moved from an RTX 4090 to an RTX 6000 Pro Blackwell. Since Blackwell requires CUDA ≥ 12.8, I tried using the NGC TensorFlow 25.02 (TF 2.17, CUDA 12.8) container to run Sionna (especially Sionna RT), but I still can’t get it to work reliably.

Is Blackwell currently supported for Sionna RT with Mitsuba 3.6.2 / Dr.Jit 1.0.3?
If yes, is there a recommended setup (versions or container) to run Sionna / Sionna RT on Blackwell GPUs? Any guidance or known workarounds would be greatly appreciated. Thanks!

[merlinND]

Hello @hssam1002,

There was an issue in DrJit 1.0.3 which prevented it from running correctly on 50-series GPUs. The fix was already implemented, but will only be included in the next Sionna release, which will come out shortly.
In the meantime, you can follow the instructions in this comment #913 (comment)

Please let me know if you encounter issues other than those mentioned in the comment above.