So, if you have other GPU hardware such as AMD, you may wish to swap it out of your Windows PC for an Nvidia GPU, and set up new Nvidia driver software. Apple has released Tensorflow Metal to enable AMD GPU for Tensorflow (see a tutorial by Jack Chih-Hsu Lin on Medium, on setting Tensorflow Metal up on a Mac), but I personally only like Windows and Linux, so I won’t be covering any Mac based tutorials. Keras) permit significantly faster training of deep learning when they are set up with GPU (graphics processing unit) support compared with using a CPU.įor GPU support to be available for those deep learning frameworks, the GPU itself must be compatible with the CUDA toolkit and any additional required GPU-Accelerated Libraries, for example cuDNN.Īt present, CUDA compatibility is largely limited to Nvidia GPUs. Frameworks such as Tensorflow, Pytorch, Theano and Cognitive Toolkit (CNTK) (and by extension any deep learning library which works alongside them, e.g.