We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 358928c commit f434dc1Copy full SHA for f434dc1
1 file changed
src/session_info2/_pu.py
@@ -4,6 +4,7 @@
4
import os
5
import platform
6
import shutil
7
+import sys
8
from multiprocessing import cpu_count
9
from pathlib import WindowsPath
10
from subprocess import CalledProcessError, run
@@ -12,7 +13,14 @@
12
13
def cpu_info() -> str:
14
"""Get CPU info."""
15
proc = platform.processor() or None
- return f"{cpu_count()} logical CPU cores{f', {proc}' if proc else ''}"
16
+ total_cores = cpu_count()
17
+ if sys.version_info >= (3, 13):
18
+ avail_cores = os.process_cpu_count()
19
+ elif platform.system() == "Linux":
20
+ avail_cores = len(os.sched_getaffinity(0))
21
+ else:
22
+ avail_cores = total_cores
23
+ return f"{avail_cores}/{total_cores} logical CPU cores{f', {proc}' if proc else ''}"
24
25
26
def gpu_info() -> tuple[str, ...]:
0 commit comments