cupy.cuda.runtime.setDevice

cupy.cuda.runtime.setDevice(int device)