yqyao/SSD_Pytorch

about cuda

wangxianghan opened this issue · 1 comments

wangxianghan commented

Thank you very much for your work, but I have encountered some questions about cuda
image

image

yqyao commented

Because your machine only have two gpu, but I set CUDA_VISIBLE_DEVICES = 3,2,1,0. So you need to change it to "1,0" in train.py (second line). @wangxianghan