Which version of tensorflow is compatible with cuda 11.4?

Hello

I am struggling to install tensorflow which is compatible with cuda 11.4 so that it can use GPU instead of CPU can anyone please suggest something regarding this.

Thanks in advance

Hi @Ankita_Das, As per test build configuration there is no tensorflow version that uses CUDA 11.4. From 2.5 to 2.11 the CUDA version supported is 11.2 and from 2.12 to 2.14 the CUDA version supported is 11.8. Thank You.

yes i saw that thanks for the reply :slight_smile:

hey can you suggest which version of tensorflow should i use , beacuase i am working on a cluster system and i can’t upgrade or downgrade cuda toolkit. cuda version is 11.4.

you can try using tensorflow-2.10