NVIDIA/tensorflow

Is TensorFlow can be used with RTX 4090?

Redhwan-A opened this issue · 3 comments

Redhwan-A commented
  • 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 commented

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 commented

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 commented

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