R2D2oid edited a comment on issue #4431: URL: https://github.com/apache/incubator-mxnet/issues/4431#issuecomment-836851353
I have a similar issue where mxnet is unable to use GPU. (note: I was able to successfully switch to CPU by simply setting `context = mxnet.cpu()` and run things without any issues!) As for the GPU issue, in my case I have cuda installed and I'm able to successfully use it with torch **outside of the virtualenv that I have setup for mxnet**. Outside of my virtualenv, `torch.cuda.is_available()` returns `True` while it returns `False` within my mxnet virtualenv. If you are also using virtualenv, this may be caused by inability of virtualenv in using cuda rather than an issue mxnet. -- This is an automated message from the Apache Git Service. To respond to the message, please log on to GitHub and use the URL above to go to the specific comment. For queries about this service, please contact Infrastructure at: us...@infra.apache.org --------------------------------------------------------------------- To unsubscribe, e-mail: issues-unsubscr...@mxnet.apache.org For additional commands, e-mail: issues-h...@mxnet.apache.org
