qq_52738695的博客

原创 RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cuda:0

首先，我在浅显的运行nlp中transformer中mask掩码时,希望代码能够跑起来，但是由于可能是环境问题，将环境重新配置了一遍，但是依然解决不了。发现所有的cude：0都是由GPU运算的，这一时间发现所有的对象都是GPU运行，但是报错说就是用了两个devices，我断开链接了/(ㄒoㄒ)/~~。最后我开始使用pycharm的自带的工具断点，开始将这几句报错的话进行仔细查看参数，终于是让我发现了问题，于是我开始去歪往去搜索，这个老外给我了启发，或许我可以检查model的参数设备。