GitHub - geyanqi/simple_multi_process

Simple MultiProcess Inferencer build upon Pytorch.

torchrun --nproc_per_node <num_of_process> cpu_env_template.py

torchrun --nproc_per_node <num_of_gpu> gpu_env_template.py

Name		Name	Last commit message	Last commit date
Latest commit History 2 Commits
.vscode		.vscode
utils		utils
.gitignore		.gitignore
README.md		README.md
cpu_task_template.py		cpu_task_template.py
gpu_task_template.py		gpu_task_template.py

Navigation Menu