怎么将torch.use_deterministic_algorithms(True) 关闭
时间: 2024-02-11 16:09:56 浏览: 165
torch.cuda.is_available()返回False解决方案
5星 · 资源好评率100%
要关闭 `torch.use_deterministic_algorithms(True)`,可以调用 `torch.use_deterministic_algorithms(False)`。这将禁用 PyTorch 中的所有确定性算法。如果你只想关闭某个算子的确定性,可以在使用该算子时将 `warn_only=True`。例如:
```
with torch.backends.cudnn.flags(warn_only=True):
# 使用 adaptive_avg_pool2d_backward_cuda 算子的代码
```
这将禁用该算子的确定性,但是不会影响其它算子的确定性。请注意,关闭确定性算法可能会影响模型的可重复性和精度,所以需要根据实际情况进行取舍。
阅读全文