我正在尝试用一个GPU(NVidia GeForce GTX 750 Ti)在我的机器上训练TensorBox ReInspect实现(https://github.com/Russell91/TensorBox/)。
当我运行train.py
脚本时:
python train.py --hypes hypes/overfeat_rezoom.json --gpu 0 --logdir output
我得到以下错误:
^{pr2}$如果我在没有GPU的机器上运行相同的代码,而只在CPU上运行,则不会发生这种情况。在
这个错误的原因是什么?有什么方法可以调试吗?在
看来问题是因为州趶是真旗。请参考此线程以获取解决方案: https://github.com/Russell91/TensorBox/issues/59#issuecomment-245566316
相关问题 更多 >
编程相关推荐