Tensorflow install-no 11.2 for Ubuntu 22.04

TensorFlow General Discussion
1

Hi

Been trying forever to install Tensorflow.

Instructions on TF website say to use CUDA-toolkit 11.2 but on the Nvidia site 11.2 is not available for Ubuntu 22.04 (only 20.04 and below).

I tried using 11.7 and got everything installed ok but it gave me a lot of issues with OOM errors when training (even with the memory reduction flag). Hoping 11.2 would work better.

So do i definitely need 11.2 and if so how do I get it?

3

Have you tried to download CUDA Toolkit 11.2 ? Thank you.

5

I see the one listed under 20.04 is actually generic and can be used with 22.04.
Problem solved.

6

Yes, that is what i was trying but if you click on Linux then x86_64 then Ubuntu there is no 22.04 option…only 20.04 18.04 and 16.4

8

Can you tell us which version of TF you work with?

To solve OOM error there are a couple of things you can try -

i) Limit gpu memory growth - You can try limiting gpu memory by two ways :

a) Turn on memory growth by calling tf.config.experimental.set_memory_growth, it allocates more memory as the process increases and demands extra memory
b) Set a hard limit on the total memory tf.config.set_logical_device_configuration(memory_limit=1024).

ii) reduce batch size for training

Thank you

10

So can i have the same problem, i need to use cuda 11.2 for tensorflow 2.11 on ubuntu 22.04, because i want to use waymo dataset, and last version is: waymo-open-dataset-tf-2-11-0 · PyPI
I need to use nvidia drivers with cuda 11.2? And cuda toolkit 11.2, to compile code, how to do it? I don’t have such driver and cuda toolkit. On ubuntu 22.04, do you have any ideas how to solve it, I spent 4 days only on this problem…