From 49156eb1d18b6314554333c4bdae5ee3e6102992 Mon Sep 17 00:00:00 2001
From: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Tue, 1 Nov 2022 12:02:05 +0100
Subject: [PATCH] Fix `is_colab()` (#9994)

@AyushExel @kalenmike

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>

Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
---
 utils/general.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/utils/general.py b/utils/general.py
index 88cefb7..5cf156d 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():