Torch enable cuda. Torch enable cuda