use mx.context.num_gpus instead of mx.test_utils.list_gpus in MF recommender example

larroy · larroy · commit 6ed38bae39f9 · 2019-05-13T17:47:25.000-07:00
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"
    ]
   },

Original file line number	Diff line number	Diff line change
`@@ -55,7 +55,7 @@`
`55`	`55`	`},`
`56`	`56`	`"outputs": [],`
`57`	`57`	`"source": [`
`58`		`- "ctx = [mx.gpu(0)] if len(mx.test_utils.list_gpus()) > 0 else [mx.cpu()]\n",`
	`58`	`+ "ctx = [mx.gpu(0)] if mx.context.num_gpus() > 0 else [mx.cpu()]\n",`
`59`	`59`	`"batch_size = 128"`
`60`	`60`	`]`
`61`	`61`	`},`