about cuda - Githubissues

yqyao / SSD_Pytorch

support different SSDs and different scale test, support refineDet.

MIT License

148 stars 51 forks source link

about cuda #4

Closed wangxianghan closed 5 years ago

wangxianghan commented 5 years ago

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

yqyao commented 5 years ago

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