check for kaggle env variable

Lightning-AI · Apr 23, 2020 · 5ca9dbb · 5ca9dbb
1 parent 3c6f856
commit 5ca9dbb
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pytorch_lightning/trainer/trainer.py b/pytorch_lightning/trainer/trainer.py
@@ -751,7 +751,7 @@ def fit(
             log.info(f'training on {self.num_tpu_cores} TPU cores')
 
             #  COLAB_GPU is an env var available by default in Colab environments.
-            start_method = 'fork' if os.getenv('COLAB_GPU') else 'spawn'
+            start_method = 'fork' if os.getenv('COLAB_GPU') or os.getenv('KAGGLE_URL_BASE') else 'spawn'
 
             # track for predict
             self.model = model