cpuset_py3_create_configure_and_assign_process_to_cpuset.py

1import os
2from cpuset import Cpusets
3
4# Initialize the cpuset manager
5# By default, this interacts with /sys/fs/cgroup/cpuset
6cs = Cpusets()
7
8# Define the name for a new cpuset
9set_name = "example_set"
10
11# Create a new cpuset if it doesn't exist
12if set_name not in cs:
13    cs.create(set_name)
14
15# Configure the cpuset
16# Assign CPUs 0 and 1 to this set
17cs[set_name].cpus = "0-1"
18# Assign Memory Node 0 (common for single-socket systems)
19cs[set_name].mems = "0"
20
21# Move the current process (PID) into the new cpuset
22current_pid = os.getpid()
23cs[set_name].tasks = current_pid
24
25print(f"Process {current_pid} moved to cpuset '{set_name}'")
26print(f"CPUs assigned: {cs[set_name].cpus}")
27
28# Note: In production, ensure you have sufficient permissions (sudo) 
29# to write to cgroup files.