opened
indhub
username_0
Use GPU if available. 

Replace 

```
ctx = mx.cpu()
```

with 

```
ctx = mx.gpu() if mx.test_utils.list_gpus() else mx.cpu()
```
Changes to test notebooks from CI
issue
created
vishaalkapoor
username_1
I verified that the notebooks in this PR have already been updated to use the gpu. Closing out this PR.
comment