I saw that it was not working with google colab free version. But it is now working after I upgraded it to colab pro. May be the colab free version lacks some functionality that’s provided by the pro version.
(May be the gpu in free version is a problem, because as per the google, colab pro provides better gpu, larger disk space and longer runtimes. There’s no mention about any other special functionality provided by the pro version.)