pytorch supports gpu acceleration