chinakook edited a comment on issue #19649:
URL:
https://github.com/apache/incubator-mxnet/issues/19649#issuecomment-751424572
The result also varies in mxnet_cu110-2.0.0b20201226
minimum test case to reproduce that:
```python
import os
# os.environ['MXNET_CUDNN_AUTOTUNE_DEFAULT'] = '0'
import mxnet as mx
import numpy as np
from mxnet.gluon.model_zoo.vision.resnet import resnet18_v1
def testrestnet():
ctx = mx.gpu(0)
mx_model = resnet18_v1(pretrained=False)
mx_model.hybridize()
mx.random.seed(22)
mx_model.initialize()
mx_model.reset_ctx(ctx=ctx)
np.random.seed(115)
x = np.random.uniform(size=(1,3,224,224)).astype(np.float32)
x_mx = mx.nd.array(x, ctx=ctx)
y_mx = mx_model(x_mx)
# the res is -1219.706 on RTX3090 with MXNET_CUDNN_AUTOTUNE_DEFAULT=0
# the res varies on RTX3090 GPU without MXNET_CUDNN_AUTOTUNE_DEFAULT=0:
-1219.7754, -1220.0055, -1220.0052, -1220.0051
# the res is -1220.0052 on RTX2080Ti
# the res is -1220.0062 on CPU
res = y_mx.asnumpy().sum()
print(res)
if __name__ == '__main__':
testrestnet()
```
----------------------------------------------------------------
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]
