Hi, we have a problem with setting up an environment where we can utilize tensorflow (with gpu suport) on Ubuntu 20.04.
We have tried different guides and versions of CUDA, now we use nvidia-driver-460, cuDNN7.6.5, toolkit10.1, python 3.8.5 and tensorflow=2.5.0-dev20201028.
We have also tried to upgrade to driver-455, cuDNN8.0, toolkit 11.1 and had no more sucess.
Even though the GPU is found by tensorflow it fails to use it properly. importing packages, and show images work fine but when we try to load a custom model it freeze. No real error appear and the GPU is allocating 22/24 GB memory but nothing is happening.
All of this worked fine on a geforce1050 with the same code.
What can be the issue? What are the exact versions of driver, cuDNN, toolkit, python and tensorflow to make this work? If we can get the correct versions we can begin installing those and send specific errors for those versions.
Really hope someone has the time to help us with this, we have been stuck for several days now.
Best regards, Jonatan and Marcus