Six-walnuts-Lu opened this issue 3 years ago · 1 comments
Specify a single GPU
CUDA_VISIBLE_DEVICES=1 python train.py