定位到报错那一行,在前一行加入
torch.use_deterministic_algorithms(False)
这个报错信息是关于使用PyTorch的upsample_bilinear2d_backward_out_cuda
函数时的一个警告。该警告指出该函数没有确定性的实现,但设置了torch.use_deterministic_algorithms(True)
来启用确定性算法。
PyTorch中的确定性算法通常用于确保在相同的输入数据和计算设备下,同样的操作能够产生相同的结果。然而,并非所有操作都有确定性的实现。在这种情况下,PyTorch会发出警告。
该警告提供了两种解决方法:
- 对于此操作,可以关闭确定性。通过执行
torch.use_deterministic_algorithms(False)
来禁用确定性算法。 - 如果只是想在遇到不确定性时发出警告,而不是禁用整个确定性算法,可以在使用该函数时添加
warn_only=True
选项,如:upsample_bilinear2d_backward_out_cuda(..., warn_only=True)
。