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]

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