解决方法:将ssd_pascal.py文件中gpus = "0,1,2,3"的GPU选择改为gpus = "0",后面的1,2,3都删掉即可。因为电脑只安装了一块GPU。还需要将ssd_pascal.py文件中中的batch_size = 32和accum_batch_size = 32同时改为8,不然会出现“Check failed: error == cudaSuccess (2 vs. 0) out of memory”的错误。
配置SSD-caffe测试时出现“Check failed: error == cudaSuccess (10 vs. 0) invalid device ordinal”解决
最新推荐文章于 2020-07-04 20:02:40 发布