diff --git a/source/tests/pt/__init__.py b/source/tests/pt/__init__.py
index 6ceb116d85..fdbdd73f79 100644
--- a/source/tests/pt/__init__.py
+++ b/source/tests/pt/__init__.py
@@ -1 +1,5 @@
 # SPDX-License-Identifier: LGPL-3.0-or-later
+import torch
+
+torch.set_num_threads(1)
+torch.set_num_interop_threads(1)