Code runs on “cpu”. My Pytorch vesion is 1.4.0.
As you recommended i have installed the nightly version of pytorch 1.7.0 nightly (gpu version) in a new environment and also installed DGL (gpu version). But error remains the same.