chinakook commented on issue #19649:
URL: 
https://github.com/apache/incubator-mxnet/issues/19649#issuecomment-751439240


   A torch test case. Torch has lower difference between 2080Ti and 3090. 
However, MXNet will have difference up to 0.3 in some cases.
   ```python
       import torch 
       import torchvision as tv
       torch.backends.cudnn.benchmark=True
   
       model = tv.models.resnet18(pretrained=True)
       model.cuda(0)
       model.eval()
   
       # y is always 948.1921 on CPU
       # y is always 948.1919 on RTX2080Ti whenever cudnn.benchmark is True or 
False
       # y is 948.19165 on RTX3090 when cudnn.benchmark=False
       # y varies on RTX3090 when cudnn.benchmark=True: 948.19147, 948.1919
       x = torch.ones(1,3,224,224).cuda(0)
       y = model(x)
       y = y.abs().sum()
       print(y.detach().cpu().numpy())
   ```


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

Reply via email to