AssertionError：assert all(map(lambda i: i.is_cuda, inputs))

出错原因：

单GPU跑模型时，机器上存在多个GPU，程序运行不知道使用哪一个gpu

解决办法：

指定GPU方法一：

model = torch.nn.DataParallel(model,device_ids=[0])

指定GPU方法二：

import os
os.environ[“CUDA_VISIBLE_DEVICES”] = "0" 

参考：

assert all(map(lambda i: i.is_cuda, inputs)) AssertionError_机器不学习我学习的博客-CSDN博客