diff --git a/docs/tutorials/gluon/info_gan.md b/docs/tutorials/gluon/info_gan.md
index 55e9660e23aa..88ab572a60fd 100644
--- a/docs/tutorials/gluon/info_gan.md
+++ b/docs/tutorials/gluon/info_gan.md
@@ -33,7 +33,7 @@ batch_size   = 64
 z_dim        = 100
 n_continuous = 2
 n_categories = 10
-ctx      = mx.cpu()
+ctx = mx.gpu() if mx.test_utils.list_gpus() else mx.cpu()
 ```
 
 Some functions to load and normalize images.