在学习过程中遇到一个问题,就是当我想把一个GPU中的数据通过python中的serial模块传出来时,报错,并显示说我不能把GPU里面的数据拿出来,需要先将其转移到CPU中,当我按照他的提示使用 变量.CPU() 还是报错,提示我list没有 .cpu 属性。
解决办法:使用torch.tensor()这个函数
例如:L是一个在GPU中的list(列表)。
torch.tensor(L, device=‘cpu’) 就可以将list转换为tensor,并将数据转移到cpu中。
自己的是这么做的,肯定还有其他更好的方法,大佬路过多多指点。第一次写博客,哪里写不好了,多多指教。