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

…mmender example
apache · May 14, 2019 · 6ed38ba · 6ed38ba
1 parent 13d6ee6
commit 6ed38ba
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"
    ]
   },