How to install GPU drivers for GeForce RTX 3090?

General Discussion
,
#1

Hi, we want to run TensorFlow on an Ubuntu 22.04 system, with Intel i9-10980XE cpu and 2 x GeForce RTX 3090 GPU.

This has worked in the past, but the drivers are no longer working since upgrading from Ubuntu 20.04. Therefore we are thinking of doing a clean o/s and drivers install.

Installation of the GPU drivers seems quite complex. I don’t understand the difference between the NVIDIA drivers, CUDA toolkit and cuDNN SDK. Version numbers seem critical. The TensorFlow installation instructions refer to the NVidia site but there doesn’t seem to be one overall set of instructions.

Please can anyone provide us with some general advice on the installation or point us to some other instructions?

1 Like
#3

Hi @David_A, For using tensorflow with GPU you can use pip install tensorflow[and-cuda] command which will automatically install cuda and cudnn packages supported for tensorflow version 2.15. Thank You.

1 Like
#4

Thank you! This helped me put things a bit more into context with tensorflow+cuda (pip install).

  • David
    Over and Out
#5

Thank you for your answer.

1 Like