Tensorflow GPU in Ubuntu 22.04

[sgvignali]

Hi all,

would it be possible for you to update the installation steps for ubuntu 22.04? This would be great!

Thank you in advance,

Sergio

[t-kalinowski]

Unfortunately, configuring TensorFlow to use GPUs on Ubuntu 22.04 today requires jumping through some hoops or taking on some binary-incompatibility risks.

The current status:

• Official prebuilt binaries of TensorFlow wheels are built to use cuda 11.2.
• The latest CUDA installer .deb's from NVIDIA with prebuilt binaries of CUDA 11.2 are for Ubuntu 20.04: https://developer.nvidia.com/cuda-11.2.2-download-archive?target_os=Linux&target_arch=x86_64&target_distro=Ubuntu. Installing a deb built for Ubuntu 20.04 on Ubuntu 22.04 may work, or it may not (I haven't tested).