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