TNTran92 commented on issue #18879:
URL:
https://github.com/apache/incubator-mxnet/issues/18879#issuecomment-1125431990
I am having the same issue. mxnet runs just fine when ctx is cpu (or
default) but stalls as soon as gpu is specified.
`----------Python Info----------
Version : 3.7.0
Compiler : MSC v.1912 64 bit (AMD64)
Build : ('default', 'Jun 28 2018 08:04:48')
Arch : ('64bit', 'WindowsPE')
------------Pip Info-----------
Version : 21.2.4
Directory :
C:\Users\trana\anaconda3\envs\gluon-ts-gpu\lib\site-packages\pip
----------MXNet Info-----------
Version : 1.5.0
Directory :
C:\Users\trana\anaconda3\envs\gluon-ts-gpu\lib\site-packages\mxnet
Num GPUs : 1
Hashtag not found. Not installed from pre-built package.
----------System Info----------
Platform : Windows-11
----------Hardware Info----------
machine : AMD64
processor : AMD64 Family 25 Model 33 Stepping 2, AuthenticAMD
Name
AMD Ryzen 9 5900X 12-Core Processor `
I am using mxnet-cu101 with CUDA 10.1 and CuDNN 8.0.5.39.
GPU RTX3060 driver version 512.77
##############################
The code looks like this
>>> from mxnet import nd, gpu, gluon, autograd
>>> from mxnet.gluon import nn
>>> from mxnet.gluon.data.vision import datasets, transforms
>>> import time
>>> x = nd.ones((3,4))
>>> x
[[1. 1. 1. 1.]
[1. 1. 1. 1.]
[1. 1. 1. 1.]]
<NDArray 3x4 @cpu(0)>
>>> x = nd.ones((3,4), ctx=gpu())
##### Stalls right here#####
--
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.
To unsubscribe, e-mail: issues-unsubscr...@mxnet.apache.org
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
