bytedance/lightseq

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.

skill-diver opened this issue · 0 comments

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