UBLAS_WORKSPACE_CONFIG to :16:8 to only use one buffer size during runtime.
PyTorch typically benchmarks multiple algorithms to select the fastest one, but if you want reproducibility, you should disable this feature because the benchmark may select different algorithms each time. Lastly, pass True to torch.use_deterministic_algorithms to enable deterministic algorithms.


	Copied
import os

os.environ["CUBLAS_WORKSPACE_CONFIG"] = ":16:8"

torc