pytorch 多卡训练示例
import torch
import torch.nn as nn
from torch.utils.data import DataLoader,Dataset
# parameter and DataLoaders
input_size = 5
output_size = 2
batch_size = 30
data_size = 100
device = torch.device("cuda:0" if torch.cuda.is_available() else"cpu")
class RandomDataset(Dataset):
def __init__