Move Torch package addition to device specific confics, (2/2)