lfz / DSB2017

The solution of team 'grt123' in DSB2017

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

GPU issue

qitao052 opened this issue · comments

Thanks a lot for this project.
I just downloaded this code project and run main.py for my own data just for prediction. I configured the configure_submit.py with n_GPU =2 and run Python main.py. The only problem is my GPU is not used and the calculation is very slow. I used nvidia-smi to show the usage of my GPUs and no process is using GPUs. Do you have any suggestions to figure out the problem? Thanks a lot