Exception: Installed CUDA version 12.3 does not match the version torch was compiled with 12.1, unable to compile cuda/cpp extensions without a matching cuda version.

bytedance / lightseq

LightSeq: A High Performance Library for Sequence Processing and Generation

Other

3.21k stars 329 forks source link

Exception: Installed CUDA version 12.3 does not match the version torch was compiled with 12.1, unable to compile cuda/cpp extensions without a matching cuda version. #526

Open skill-diver opened 9 months ago

skill-diver commented 9 months ago

I have this error when run, how could I find compatible pytorch?

Exception: Installed CUDA version 12.3 does not match the version torch was compiled with 12.1, unable to compile cuda/cpp extensions without a matching cuda version.

I'm now using CUDA 12.3, and pytorch 2.2