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:

  1. There are no previous nvidia drivers on the machine. If there are, I recommend to uninstall them first otherwise following my instructions may or may not work.
  2. Python is already installed on the machine. I recommend using anaconda since it makes the job of tracking library versions and virtual environments very easy.

#cuda #tensorflow #pytorch #machine-learning #deep-learning #painless gpu and tensorflow/pytorch integration on ubuntu 18.04

Painless GPU and tensorflow/PyTorch integration on Ubuntu 18.04
1.45 GEEK