Support for 5090?

[stav-ros]

Hi, I was wondering if there's a way to upgrade torch and cuda to 12.6 for support of blackwell.

[abartesaghi]

The release candidate 0.7.3 should already support Blackwell cards. See this issue. If you are already using that version and still having problems, we will need to know more about the specific program that is failing to run.

[stav-ros]

I've upgraded to 0.7.3, however it is topaz denoising that still gives the error message.

UserWarning: 
19	[INFO] NVIDIA GeForce RTX 5090 with CUDA capability sm_120 is not compatible with the current PyTorch installation.
20	[INFO] The current PyTorch install supports CUDA capabilities sm_50 sm_60 sm_70 sm_75 sm_80 sm_86 sm_90.
21	[INFO] If you want to use the NVIDIA GeForce RTX 5090 GPU with PyTorch, please check the instructions at https://pytorch.org/get-started/locally/
22	[INFO] # using CUDA device: 0
23	[INFO] # denoising with patch size=128 and padding=48
24	[INFO] Traceback (most recent call last):
25	[INFO]   File "/opt/conda/envs/topaz/bin/topaz", line 8, in <module>
26	[INFO]     sys.exit(main())
27	[INFO]   File "/opt/conda/envs/topaz/lib/python3.8/site-packages/topaz/main.py", line 148, in main
28	[INFO]     args.func(args)
29	[INFO]   File "/opt/conda/envs/topaz/lib/python3.8/site-packages/topaz/commands/denoise3d.py", line 768, in main
30	[INFO]     denoise(model, path, args.output, args.suffix
31	[INFO]   File "/opt/conda/envs/topaz/lib/python3.8/site-packages/topaz/commands/denoise3d.py", line 667, in denoise
32	[INFO]     x = (x - mu)/std
33	[INFO] RuntimeError: CUDA error: no kernel image is available for execution on the device
34	[INFO] CUDA kernel errors might be asynchronously reported at some other API call, so the stacktrace below might be incorrect.
35	[INFO] For debugging consider passing CUDA_LAUNCH_BLOCKING=1
36	[INFO] Compile with `TORCH_USE_CUDA_DSA` to enable device-side assertions.
37	[INFO] Converting tomogram to 16-bits
38	[INFO] nextPYP (topaz denoising) finished successfully

[abartesaghi]

Thanks for reporting this. We updated the python/cuda/topaz versions to support Blackwell GPUs. If you reinstall 0.7.3, you will get the new updates.