Setup your remote Ubuntu 20.04 with cuda for deep learning with pytorch

I do this guide as a note for me and maybe also for other people. I take this note to remember the step to do on a remote ssh server toprepare it for deep learning projects.

Pre requirements

GCC if you do not have it:

Install nvidia driver

It can be installed also automatically with cuda, but is not a bad idea to controll it manually.

Install Cuda 10.2

I follow this guide.

Install CuDNN

Download it from this link and after scp it to the remote machine

Install using this guide:

Install python3 and pip3

Install pytorch and test gpu support

If you use a particular version of cuda you canobtain the correct pip install on this site. Generally we can do the following:

Install Tensorflow and testt gpu support