enforce deterministic GPU algorithms

[cameronraysmith]

See torch.use_deterministic_algorithms and CUDA cublas results reproducibility.

import os
os.environ["CUBLAS_WORKSPACE_CONFIG"] = ":4096:8"
print(os.environ["CUBLAS_WORKSPACE_CONFIG"])

import numpy as np
import torch

torch.use_deterministic_algorithms(True)

SEED = 1

np.random.seed(seed=SEED)
torch.manual_seed(SEED)