Using CUDA in Python

📝 The code that involves CUDA will run on Google colab only. Have a look at Appendix 1 to learn the instructions for working in the colab workspace.

To use Google’s GPUs from the Python Notebook’s we need to change a setting. From the menu option at the top of our notebook choose Runtime and then Change Runtime type.