Cuda version help

Env - WSL 2 with Ubuntu

I am installing a python lib using pip = pip install flash-attn.
Its giving error - needs cuda version11.6 or above
I checked my cuda version using nvcc- V , its 11.5 but with nvida smi, its 12.4.

Attaching the lib vesions of virtual env, can you guys please help me what do to?

lib versions of virtual env, this also contain nvidia drivers
requirements_3.10.txt (944 Bytes)

Installation error

Your current Cuda version is 11.5 so just updating to 11.6 or higher should work. The 12.4 from nvidia-smi is the version of Cuda your driver is good up to and will be fine with the older toolkit, just skip the driver install during the toolkit update.

1 Like

Thanks for the response I upgraded to 11.8.