Hi thanks for your answer. We tested this on two different machines and we get the same result.
Machine 1:
GPU: GTX 1070
Pytorch: 1.1.0
Cuda: 9.0.36
Driver: 430.14
Machine 2:
GPU: GTX 1050
Pytorch: 1.0.1
Cuda: 9.0
Driver: 430.86
It’s weird that you don’t find the same. Could it be related to CUDA 9 vs 10?