分类：两个GPU

2021
07-26

pytorch 两个GPU同时训练的解决方案

使用场景我有两个GPU卡。我希望我两个GPU能并行运行两个网络模型。代码错误代码1：#对于0号GPUos.environ['CUDA_VISIBLE_DEVICES']='0,1'device=torch.device("cuda:0"iftorch.cuda.is_available()else"cpu")#对于1号GPUos.environ['CUDA_VISIBLE_DEVICES']='0,1'device=torch.device("cuda:1"iftorch.cuda.is_available()else"cpu")0号GPU不报错，1号GPU报错。错误如下RuntimeError:Expectedtensorforargument#1... 继续阅读 >