Jiacheng Bao

icon company Shanghai AI Lab icon location Shanghai, China icon location I am currently a PHD student at Shanghai AI Lab and NWPU, having graduated from ShanghaiTech University.

Results 1 issues of Jiacheng Bao

GPU numbers cannot be specified for multiple GPUs

1 comment

For single GPU training, I can export CUDA_VISIBLE_DEVICES, and train with headless on the specific device. But on multi-GPU training, if I don't use `export CUDA_VISIBLE_DEVICES`, and write in --devices...