fix #22

erpalma · Jul 30, 2018 · 2eca077 · 2eca077
1 parent 6c38278
commit 2eca077
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/lenovo_fix.py b/lenovo_fix.py
@@ -38,8 +38,8 @@
 
 
 def writemsr(msr, val):
-    n = glob.glob('/dev/cpu/[0-9]*/msr')
-    if not n:
+    n = ['/dev/cpu/{:d}/msr'.format(x) for x in range(cpu_count())]
+    if not os.path.exists(n[0]):
         try:
             subprocess.check_call(('modprobe', 'msr'))
         except subprocess.CalledProcessError: