For those failing when using cuda 12.9 | try `pytorch nightly cu128`

[someonenameguy]

Try using the pytorch nightly cu128.
That fixed the missing torchvision::nms error for me.

[oskeith]

Just wanted to thank you for this! I was having the torchvision::nms error with:

• cuda 11.8 + pytorch cu118 stable
• cuda 12.8.1 + pytorch cu128 stable

and this (cuda 12.8.1 + pytorch nightly/cu128) worked!

[IgnacioLuxP]

Alternatively (from PyTorch 2.8.0):

install torch==2.8.0 torchvision==0.23.0 torchaudio==2.8.0 --index-url https://download.pytorch.org/whl/cu129

Worked for me with CUDA Version 12.9 and the latest mokuro version (0.2.2).