I would like to run a trained model on a specific GPU. I tried torch.nn.DataParallel(model, device_ids=[2]) and make all xx.cuda(device=2). However, it always uses GPU memory from GPU 0. I am using pytorch 0.3.1. Any suggestions to fix it? Thanks!