Limit the usage of CPUs and GPUs of a process

CUDA_VISIBLE_DEVICES=0,1 taskset --cpu-list 0,1,2,3 python train.py