diff --git a/utils/general.py b/utils/general.py
index 88cefb7bb..5cf156dfe 100644
--- a/utils/general.py
+++ b/utils/general.py
@@ -72,7 +72,7 @@ def is_chinese(s='人工智能'):
 
 def is_colab():
     # Is environment a Google Colab instance?
-    return 'COLAB_GPU' in os.environ
+    return 'google.colab' in sys.modules
 
 
 def is_notebook():