‘list’object has no attribute ‘cuda’

想把代码放到GPU上跑，将数据集转移到CUDA上出现错误‘list’object has no attribute ‘cuda’无脑百度原因是list列表没有CUDA属性无脑通过numpy转化为tensor 后报错ValueError: only one element tensors can be converted to Python scalars仔细看了看，自己数据是这样的 [tensor

小菜狗很无敌

5266人浏览 · 2021-12-04 16:45:53

小菜狗很无敌 · 2021-12-04 16:45:53 发布

想把代码放到GPU上跑，将数据集转移到CUDA上出现错误‘list’object has no attribute ‘cuda’
无脑百度
原因是list列表没有CUDA属性
无脑通过numpy转化为tensor 后
报错ValueError: only one element tensors can be converted to Python scalars

仔细看了看，自己数据是这样的 [tensor,tensor,tensor]
因此
targets =[ target.cuda() for target in targets]
解决
很简单，希望自己以后遇到问题先动脑子

Triton中文社区

欢迎来到由智源人工智能研究院发起的Triton中文社区，这里是一个汇聚了AI开发者、数据科学家、机器学习爱好者以及业界专家的活力平台。我们致力于成为业内领先的Triton技术交流与应用分享的殿堂，为推动人工智能技术的普及与深化应用贡献力量。

更多推荐

torch.device(‘cuda:0‘ if cuda else ‘cpu‘)

# 这个地方的cuda:0 实际上并不是0号GPU，他取决于CUDA_VISIBLE_DEVICES# 然后逻辑GPU和物理GPU有一个对应关系# 如果CUDA_VISIBLE_DEVICES为2,1,3# 那么CUDA:0就是2号GPU， CUDA:1 就是1号GPU CUDA:3 就是3号GPUreturn torch.device('cuda:0' if cuda else 'cpu')..