'Newbie: Wan 2.1 Error: Unsupported CUDA architecture: sm120'

Hı everyone
I have rtx 5090
ryzen 7 9800x3d
win11
96gb ram
trying to run wan 2.1 with the torch files which works fine with stable diffisuon but i am having
‘Unsupported CUDA architecture: sm120’ error, can anyone help to fix it?

i installed cuda toolkit 12.8 and also updated lastest torchs in the env folder of wan 2.1 (pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118)

But still have the same error.

Hi @Moodrabbit what version of CUDA do you have installed? you may need to update it.

@Moodrabbit - You don’t need to install CUDA with AI Workbench.

It’s handled at the container level.

If you aren’t using Workbench, then you might be better off posting in this forum: Latest Deep Learning (Training & Inference)/Frameworks topics - NVIDIA Developer Forums

1 Like

i installed cuda toolkit 12.8 and also updated lastest torchs in the env folder of wan 2.1

Seems like you aren’t using Workbench.

Can you please post in an appropriate forum?

1 Like