How to run Jupyter Notebook on GPU


I was able to launch the GPU instance and install everything I need. I tested it with Python codes given by MXNet install instructions on AWS DL AMI and it printed out the expected result. However, after I opened the Jupyter notebook and ran the same test. It gave me the error “Operator _ones is not implemented for GPU.” Any thoughts or ideas? Thank you very much for your help!


Resolved.Thanks you.


I was wondering how you resolve this? I have the same issue.