modify script, move gpu_id as input argement