点击关注我的GitHub
2022年7月29日
Pytorch多GPU并行训练模型
摘要： Pytorch多GPU并行训练模型 如果要用多GPU训练模型，那就要用到pytorch中的nn.DataParallel()函数对模型进行封装。 1 # code_for_use_cuda 2 device_count = torch.cuda.device_count()#获得本机的显卡的个数 3 print("cuda.devic 阅读全文
posted @ 2022-07-29 15:51 justkeen 阅读(630) 评论(0) 推荐(0) 编辑