New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

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

Open

cdycdycdy opened this issue Aug 27, 2021 · 1 comment

cdycdycdy commented Aug 27, 2021

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

yangyuqing15715165798 commented Sep 1, 2021

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

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

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment