Pytorch cuda (6/10)

Pytorch cuda