I would like to figure out, whether the PyTorch model is on cpu or cuda in ... to train VGG16 by using torch RPC. device_count ()) only one GPU is detected.
確定! 回上一頁