chinakook edited a comment 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 on RTX3090 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:
[email protected]
---------------------------------------------------------------------
To unsubscribe, e-mail: [email protected]
For additional commands, e-mail: [email protected]
