Remove `set_device`
Remove `set_device`

See https://pytorch.org/docs/stable/cuda.html#torch.cuda.set_device

undertherain/benchmarker,undertherain/benchmarker,undertherain/benchmarker,undertherain/benchmarker