GPU下运行
yuan-self opened this issue · comments
你好请问一下按照你的代码用GPU跑的话,最后跑的时候怎么还是用CPU再运行呀
@goudi520 你好,请确认如下事项:
- 你能用pytorch环境正常在GPU上训练其他任务
- 启动脚本中有
--gpu 0
项,且选择的是某个可用的GPU id
这个库就是基于pytorch的基础API的,具体的device代码请参见CRNet/utils/init.py
中的init_device
函数,应该问题不大
好的谢谢你的回复 我再去看一看代码