diff --git a/examples/notebooks/a_gentle_introduction_to_tsl.ipynb b/examples/notebooks/a_gentle_introduction_to_tsl.ipynb
index b59b3c9..98159b7 100644
--- a/examples/notebooks/a_gentle_introduction_to_tsl.ipynb
+++ b/examples/notebooks/a_gentle_introduction_to_tsl.ipynb
@@ -1023,7 +1023,7 @@
     "\n",
     "trainer = pl.Trainer(max_epochs=100,\n",
     "                     logger=logger,\n",
-    "                     gpus=1 if torch.cuda.is_available() else None,\n",
+    "                     devices=1 if torch.cuda.is_available() else None,\n",
     "                     limit_train_batches=100,  # end an epoch after 100 updates\n",
     "                     callbacks=[checkpoint_callback])\n",
     "\n",