各位好：如何在train.py中的main 中添加指定使用GPU的代码，代码默认跑cpu

chineseocr / table-detect

table detect(yolo) , table line(unet)

MIT License

236 stars 86 forks source link

Open cdycdycdy opened 3 years ago

cdycdycdy commented 3 years ago

os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID" os.environ["CUDA_VISIBLE_DEVICES"] = "0" 无效果，用的是pubtabnet数据集

yangyuqing15715165798 commented 3 years ago

os.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID" os.environ["CUDA_VISIBLE_DEVICES"] = "0" 无效果，用的是pubtabnet数据集

你有处理好pubtable的json格式吗？这个项目好像只读json格式的