Torch enable cuda (6/10)

Torch enable cuda