Using device = cuda:0