Pytorch cuda (1/10)

Pytorch cuda