use mx.context.num_gpus instead of mx.test_utils.list_gpus in MF reco…

…mmender example (apache#14926)
haohuanw · Jun 23, 2019 · 0f11f99 · 0f11f99
1 parent c327464
commit 0f11f99
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/example/recommenders/demo1-MF.ipynb b/example/recommenders/demo1-MF.ipynb
@@ -55,7 +55,7 @@
    },
    "outputs": [],
    "source": [
-    "ctx = [mx.gpu(0)] if len(mx.test_utils.list_gpus()) > 0 else [mx.cpu()]\n",
+    "ctx = [mx.gpu(0)] if mx.context.num_gpus() > 0 else [mx.cpu()]\n",
     "batch_size = 128"
    ]
   },