Need support more new versions of CUDA

[donhuvy]

Bug description

Need support more new versions of CUDA.

Reproduced in studio

No response

What operating system are you using?

Unknown

LitGPT Version

No response

[lianakoleva]

This bug is due to the version of PyTorch on your system. Resolve it by pip install torch>=2.7.

[Borda]

Why just use pip install torch -U?

[donhuvy]

I don't really understand deeply problem. But now I run, then ouput is ok. Maybe new PyTorch helps resolving problem.

[Borda]

You may also install torch for a specific cuda version... By adding index option to pip, for example pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121