Use torch.set_default_device in conftest.py
#232
Merged
torch.set_default_device in conftest.py
#232