Run Tensorflow on RTX 5080

TensorFlow
,
1

Hello Developers,

I have NVIDIA GeForce RTX™ 5080 graphics card, I’m using Ubuntu 24.04.3 LTS OS.
Can you please provide the steps so that I can use GPU acceleration for Tensorflow or redirect to appropriate resources.

Regards,
Ankit

3

Hi @dsp_rtl , A warm welcome to the Google AI Forum!
To get started with TensorFlow GPU follow these installation steps.

  1. Install Nvidia Graphics Card & Drivers
  2. Download & Install CUDA
  3. Download & Install cuDNN
  4. Install tensorflow-gpu and verify the installation
  5. Verify the installation by executing the following command:
 python3 -c "import tensorflow as tf; print(tf.config.list_physical_devices('GPU'))"

Kindly refer to the official TensorFlow GPU document.
Thank you!