[cudnn_frontend] Error: No execution plans support the graph (--Solved--)

[AfterHAL]

I did a fresh install od Ubuntu and ai-toolkit today and got this error running the script :

RuntimeError: cuDNN Frontend error: [cudnn_frontend] Error: No execution plans support the graph

This is because standard installation installs pytorch 2.5.0 (since Oct 23, 2024).

Just downgrade pytorch to 2.4.1 :

pip3 install torch==2.4.1 torchvision==0.19.1 torchaudio==2.4.1 --index-url https://download.pytorch.org/whl/test/cu124