If runtime type is set to GPU in Google colab, then we are able to use GPU only for large computations such as neural network training etc. But I want to use the G