Hey,
I’m not sure if this will be helpful or not but if you use pytorch 0.3.1 you can direct your model to run on a specific gpu by using model.cuda(_GPU_ID) #_GPU_ID should be 0, 1, 2 etc
if you are using pytorch 0.4 you can direct your model to run on a specific gpu by using
device = torch.device("cuda:1")
model.to(device)