assert all(map(lambda i: i.is_cuda, inputs)) AssertionError 报错,原因是pytorch在服务器上跑使用了多个gpu,暂时不知道怎么用多个gpu,但是为了跑通加入了指定单一gpu的代码: import os os.environ[“CUDA_VISIBLE_DEVICES”] = “1” 之后就不再报错了 欢迎留言,提出如何使用多个gpu而不会报错的方法