在终端运行:
python trainval_net.py --cuda
在pycharm中
parser.add_argument('--cuda', dest='cuda', help='whether use CUDA', action='store_true')
修改成:
parser.add_argument('--cuda', dest='cuda', default=True,
help='whether use CUDA')
最后
以上就是细心乌龟最近收集整理的关于RuntimeError: Not implemented on the CPU的全部内容,更多相关RuntimeError:内容请搜索靠谱客的其他文章。
本图文内容来源于网友提供,作为学习参考使用,或来自网络收集整理,版权属于原作者所有。
发表评论 取消回复