Is TensorFlow can be used with RTX 4090?

[Redhwan-A]

• Ubuntu 20.04.
• RTX 4090 is only working CUDA Version: 12.1.
• Python version: 3.8
• GPU model and memory: 24
• TensorFlow==2.3.

Thanks for your help.

[nluehr]

To support RTX 4090, you'll need a TensorFlow that was build against CUDA 11.8 or later. You can use a recent NVIDIA NGC docker container (22.10-tf2 or later). I believe the OSS pip package for TensorFlow 2.12.0 was also built against cuda 11.8 and should work for you, but TF 2.3 will not.

[zhangvia]

To support RTX 4090, you'll need a TensorFlow that was build against CUDA 11.8 or later. You can use a recent NVIDIA NGC docker container (22.10-tf2 or later). I believe the OSS pip package for TensorFlow 2.12.0 was also built against cuda 11.8 and should work for you, but TF 2.3 will not.

can tensorflow 1.x be used with RTX 4090?

[Redhwan-A]

@zhangvia I didn't install tensorflow-gpu, I am using tensorflow-cpu. I gave up.