From 5c2e9f54b052cc8eefaf109ec8d52537d41b1e34 Mon Sep 17 00:00:00 2001
From: Guy Jacob <guy.jacob@intel.com>
Date: Wed, 17 Jul 2019 14:28:37 +0300
Subject: [PATCH] Fix for issue #323
In execution_env.log_execution_env_state, use os.cpu_count
if os.sched_getaffinity is not available
---
distiller/apputils/execution_env.py | 6 +++++-
1 file changed, 5 insertions(+), 1 deletion(-)
diff --git a/distiller/apputils/execution_env.py b/distiller/apputils/execution_env.py
index fe7d594..d97f959 100755
--- a/distiller/apputils/execution_env.py
+++ b/distiller/apputils/execution_env.py
@@ -77,7 +77,11 @@ def log_execution_env_state(config_paths=None, logdir=None, gitroot='.'):
logger.debug("Active Git branch: %s", branch_name)
logger.debug("Git commit: %s" % repo.head.commit.hexsha)
- logger.debug("Number of CPUs: %d", len(os.sched_getaffinity(0)))
+ try:
+ num_cpus = len(os.sched_getaffinity(0))
+ except AttributeError:
+ num_cpus = os.cpu_count()
+ logger.debug("Number of CPUs: %d", num_cpus)
logger.debug("Number of GPUs: %d", torch.cuda.device_count())
logger.debug("CUDA version: %s", torch.version.cuda)
logger.debug("CUDNN version: %s", torch.backends.cudnn.version())
--
GitLab