stu1130 commented on issue #14652: CUDA 10 w/ cuDNN 7.5 Support
URL: 
https://github.com/apache/incubator-mxnet/issues/14652#issuecomment-491104313
 
 
   Here are what I found
   1. The unit test failed on cuDNN 7.5.0 & 7.5.1 but work perfect on 7.4.2 & 
7.3.1. And using Tesla V100 will resolve this problem i.e. work fine on cuDNN 
7.3.1 & 7.5.0 & 7.5.1.
   2. The function that causes the error is `cudnnGetRNNWorkspaceSize` in here
   
https://github.com/apache/incubator-mxnet/blob/874fb89cd33b0e4affd7f3fb1b4ae4e09f25ef84/src/operator/rnn-inl.h#L1369
   3. I tried CUDA 10.1 with latest CUDA driver **418.67** still not working
   
   
   

----------------------------------------------------------------
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

Reply via email to