Since we have already learned about CUDA's installation and implementation, it will now be easier for us to get started on our TensorFlow and PyTorch installation procedure. Additionally, we also need cuDNN to be installed, which is the predefined deep neural network library for CUDA. To be able to download the library, you have to fill in a free registration at https://developer.nvidia.com/, which is the official web portal for the NVIDIA Developer program.
Installing TensorFlow and PyTorch for GPUs
Installing cuDNN
In this section, we are going to install cuDNN 7.4.2 for both TensorFlow and PyTorch. Follow these steps to get started:
- Download the corresponding archive from the cuDNN repository:
- Use a Terminal and...