Totally, both are correct, but the second phrase is more related to computer system. "Switch off" is a synonym of "take off" if we are taking about devices.