How to get a correct version of mmcv

[Chauncy-Cai]

Could you plz figure out the correct version of mmcv?
I have tried 1.2.0, but No module named 'mmcv.cnn.weight_init'
I have tried some old versions which have mmcv.cnn.weight_init, but it tells me that all tensors must be on devices[0]

[JialianW]

I tested both 0.2.16 and 0.5.1.