Torch cuda (2/26)

Torch cuda