search

NVIDIA / tensorflow

An Open Source Machine Learning Framework for Everyone
https://developer.nvidia.com/deep-learning-frameworks
Apache License 2.0
962 stars 144 forks source link

Is TensorFlow can be used with RTX 4090? #96

Open Redhwan-A opened 1 year ago

Redhwan-A commented 1 year ago

Thanks for your help.

nluehr commented 1 year ago

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 8 months ago

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 8 months ago

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