在PyTorch中，通过设置DataLoader的generator参数为`torch.Generator(device=cuda)`，并同时将默认张量类型改为`torch.cuda.FloatTensor`，可以确保数据加载和生成操作在GPU上进行，避免`RuntimeError`关于预期GPU设备类型但找到CPU设备的错误。
修改DataLoader的generator（问题解决）
data_loader = data.DataLoader()的参数最后加了generator=torch.Generator(device = 'cuda')
修改将torch.set_default_tensor_type
将torch.set_default_tensor_type(‘torch.FloatTensor’)改为：torch.set_default_tensor_type(‘torch.cuda.FloatTensor’)。
参考：
RuntimeError: Expected a ‘cuda‘ device type for generator but found ‘cpu‘_alex1801的博客-CSDN博客
1万+
1099
1168
5776
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
