Cuda versioning and pytorch compatibility

Hi @ptrblck , I have same issue with cuda drivers compatibility with the pytorch version. My cuda drivers is 11.4 and the ones that bundled in PyTorch is 2.0.1+cu117 so it means it is cuda 11.7. I need a suggestion whether should I downgrade my PyTorch version or install the latest cuda version?

I’m using it to train my yolov9 model and I’m running on NVIDIA GeForce RTX 2060 SUPER. The incompatibility caused my model to have RuntimeError even before the the first epoch.

Thanks in advance.

As explained before:

1 Like