lihaofd removed a comment on issue #14476: Change RNN OP to stateful URL: https://github.com/apache/incubator-mxnet/pull/14476#issuecomment-475602423 In unix-gpu, the case in Python3: MKLDNN-GPU-NOCUDNN failed, May I know the compile option of Python3: MKLDNN-GPU-NOCUDNN? Is it make USE_BLAS=mkl USE_CUDA=1 USE_CUDNN=0 USE_MKLDNN=1 USE_CUDA_PATH=[cuda path]? I build mxnet master commit 29e13b4cbffcf47d3006b9478e242683136fff82 Author: Nathalie Rauschmayr <[email protected]> Date: Thu Mar 21 19:33:24 2019 -0700 Add then run DEVICES=/CUDA_VISIBLE_DEVICES=0 DEVICES=/gpu:0 nosetests -v tests/python/gpu/test_gluon_gpu.py:test_lstm_cpu_inference it shows error like File "/home/hli32/mxnet/stateful/mxnet_gpu_upstream/python/mxnet/gluon/block.py", line 540, in __call__ out = self.forward(*args) File "/home/hli32/mxnet/stateful/mxnet_gpu_upstream/python/mxnet/gluon/block.py", line 917, in forward return self.hybrid_forward(ndarray, x, *args, **params) File "/home/hli32/mxnet/stateful/mxnet_gpu_upstream/python/mxnet/gluon/rnn/rnn_layer.py", line 239, in hybrid_forward out = self._forward_kernel(F, inputs, states, **kwargs) File "/home/hli32/mxnet/stateful/mxnet_gpu_upstream/python/mxnet/gluon/rnn/rnn_layer.py", line 270, in _forward_kernel lstm_state_clip_nan=self._lstm_state_clip_nan) File "<string>", line 145, in RNN File "/home/hli32/mxnet/stateful/mxnet_gpu_upstream/python/mxnet/_ctypes/ndarray.py", line 92, in _imperative_invoke ctypes.byref(out_stypes))) File "/home/hli32/mxnet/stateful/mxnet_gpu_upstream/python/mxnet/base.py", line 252, in check_call raise MXNetError(py_str(_LIB.MXGetLastError())) MXNetError: [22:23:02] src/operator/rnn.cu:43: RNN on GPU is only available for cuDNN at the moment.
---------------------------------------------------------------- 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: [email protected] With regards, Apache Git Services
