Skip to content

Commit 09df36a

Browse files
committed
update seed_everything for my workspace setting up
1 parent f6f11a2 commit 09df36a

1 file changed

Lines changed: 1 addition & 0 deletions

File tree

src/py_utils/utils_torch.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -37,6 +37,7 @@ def seed_everything(seed=42, verbose=False):
3737
"""
3838
random.seed(seed)
3939
os.environ["PYTHONHASHSEED"] = str(seed)
40+
os.environ["CUBLAS_WORKSPACE_CONFIG"] = ":4096:8"
4041
np.random.seed(seed)
4142
torch.manual_seed(seed)
4243
torch.cuda.manual_seed(seed)

0 commit comments

Comments
 (0)