alloc gpus 0 1 dealloc gpus 0 1