File size: 747 Bytes
463f868
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
import multiprocessing
import os
import sys

# Ensure project root is in path
sys.path.append(os.getcwd())

from ai.train_optimized import create_env


def worker(rank):
    print(f"Worker {rank} starting pid={os.getpid()}")
    try:
        env = create_env(rank)
        print(f"Worker {rank} env created: {env}")
        env.close()
    except Exception as e:
        import traceback

        traceback.print_exc()
        print(f"Worker {rank} failed: {e}")


if __name__ == "__main__":
    multiprocessing.freeze_support()
    print(f"Main process pid={os.getpid()}")

    p = multiprocessing.Process(target=worker, args=(0,))
    p.start()
    p.join()
    print(f"Main process done. Exitcode: {p.exitcode}")