There are numerous posts about setting up your GPU on the internet. I tried to follow so many of them to set up my machine, but was somehow not able to do so. There are so many variables in this process which span from nvidia driver version, cuda tool kit version, CUDnn version, python version, tensorflow version etc. The list goes on. It is also a good idea to see if your GPU supports CUDA integration here.
I assume the following:
#cuda #tensorflow #pytorch #machine-learning #deep-learning #painless gpu and tensorflow/pytorch integration on ubuntu 18.04