@pritamdamania87, Thanks for answering.
I tried with only 1 visible GPU using CUDA_VISIBLE_DEVICES=0 in my original code.
I also got the same error as you when multiple GPUs are visible.
And I still got the following error when I add device_ids=[0]
RuntimeError: expected scalar type Half but found Float
I wonder if different version of pytorch might cause the problem?
I’m currently using 1.1.0.