From 966542c2ca1b00d128594ae4afdd638d45160972 Mon Sep 17 00:00:00 2001
From: Nick Todd <nick.todd@t-s.qinetiq.com>
Date: Mon, 5 Aug 2024 15:08:31 +0100
Subject: [PATCH] #2777: Add determinism to torch backends  when seed set.

---
 src/primaite/session/environment.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/src/primaite/session/environment.py b/src/primaite/session/environment.py
index 359932c7..a12d2eb7 100644
--- a/src/primaite/session/environment.py
+++ b/src/primaite/session/environment.py
@@ -44,6 +44,10 @@ def set_random_seed(seed: int) -> Union[None, int]:
     # if torch not installed don't set random seed.
     if sys.modules["torch"]:
         th.manual_seed(seed)
+
+    th.backends.cudnn.deterministic = True
+    th.backends.cudnn.benchmark = False
+
     return seed