Support for Cuda 12.1

[ammaryasirnaich]

I have Cuda 12.1 installed but it seems the framework supports Cuda 11.8, is there any workaround instead of downgrading the Cuda version?

The detected CUDA version (12.1) mismatches the version that was used to compile
PyTorch (11.8). Please make sure to use the same CUDA versions.

[zniihgnexy]

wierd i have the opposite:

The detected CUDA version (11.8) mismatches the version that was used to compile
PyTorch (12.1). Please make sure to use the same CUDA versions.

did you solve it? if so, could you tell me how do you fix that, many thanks :)

[RFLeijenaar]

Do you use Conda to install PyTorch by any chance? In that case it might help to remove:

• cuda-toolkit==12.1 (or any other version).

When you have (in the environment description):

• pytorch-cuda==12.1

It should already install the correct cudatoolkit version. This fixed the issue for me in building another package.