Cuda Mangement - multi-process, scheduled jobs, distributed processing
Project description
cudam
Cuda Mangement - multi-process, scheduled jobs, distributed processing
task manager
task template
#!/usr/bin/env bash
while getopts g: option;do
case "${option}" in
g) GPU_ID=${OPTARG};;
esac
done
print_help(){
printf "Parameter g(GPU ID) is mandatory\n"
printf "g values - GPU ID"
exit 1
}
if [ -z "${GPU_ID}" ];then
print_help
fi
echo "start task on GPU: $GPU_ID"
# the root directory of your python script
cd ~/code/psocnn/
# the main python script accepting the gpu ID in -g argument
python3 main.py -g ${GPU_ID}
task folder structure
task manager
# start task manager
nohup cudam_task_manager.py -n 2 -s 2 -i 10 -f 300 -l 60 &
# snap gpu
cudam_snap_gpu.py -s 2 -l 60 -g 1
Project details
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
cudam-0.0.3.tar.gz
(7.3 kB
view hashes)
Built Distribution
cudam-0.0.3-py3-none-any.whl
(16.4 kB
view hashes)