export CUDA_VISIBLE_DEVICES=0 https://stackoverflow.com/questions/39649102/how-do-i-select-which-gpu-to-run-a-job-on