restrict to 1 thread

Linardos · Aug 23, 2020 · fe8ad2c · fe8ad2c
1 parent c5fb43d
commit fe8ad2c
Show file tree

Hide file tree

Showing 7 changed files with 4 additions and 254 deletions.
diff --git a/errors.txt b/errors.txt
diff --git a/gridnode.txt b/gridnode.txt
diff --git a/inference.py b/inference.py
@@ -8,6 +8,8 @@
 os.environ["TF_CPP_MIN_LOG_LEVEL"] = "3"
 
 import torch
+
+torch.set_num_threads(1)
 import configparser
 import argparse
 import syft as sy

diff --git a/node_logs.txt b/node_logs.txt
diff --git a/node_requirements.txt b/node_requirements.txt
diff --git a/node_success.txt b/node_success.txt
diff --git a/torchlib/websocket_utils.py b/torchlib/websocket_utils.py
@@ -1,4 +1,6 @@
 import torch
+
+torch.set_num_threads(1)
 import syft as sy
 import numpy as np
 from tqdm import tqdm