diff --git a/deepwalk/__main__.py b/deepwalk/__main__.py
index c331aef8cb5255e3cd884c0606a0dceb95505d03..8f9bb24f485a30773b7443fb534b2f81fcb7dfe2 100644
--- a/deepwalk/__main__.py
+++ b/deepwalk/__main__.py
@@ -23,7 +23,13 @@ import psutil
 from multiprocessing import cpu_count
 
 p = psutil.Process(os.getpid())
-p.set_cpu_affinity(list(range(cpu_count())))
+try:
+    p.set_cpu_affinity(list(range(cpu_count())))
+except AttributeError:
+    try:
+        p.cpu_affinity(list(range(cpu_count())))
+    except AttributeError:
+        pass
 
 logger = logging.getLogger(__name__)
 LOGFORMAT = "%(asctime).19s %(levelname)s %(filename)s: %(lineno)s %(message)s"