在学习DRbox过程中,用训练好的模型在测试单个图片时总是出现显存溢出的问题,请问对电脑的硬件有什么要求

Input Image Size=(193, 120, 3)
Start detection
(64, 3, 300, 300)
/home/jinsong/anaconda2/lib/python2.7/site-packages/skimage/transform/_warps.py:105: UserWarning: The default mode, ‘constant’, will be changed to ‘reflect’ in skimage 0.15.
warn("The default mode, ‘constant’, will be changed to ‘reflect’ in "
/home/jinsong/anaconda2/lib/python2.7/site-packages/skimage/transform/_warps.py:110: UserWarning: Anti-aliasing will be enabled by default in skimage 0.15 to avoid aliasing artifacts when down-sampling images.
warn("Anti-aliasing will be enabled by default in skimage 0.15 to "
Pre-processing time = 0.00421500205994
F0326 17:11:08.557667 13540 syncedmem.cpp:56] Check failed: error == cudaSuccess (2 vs. 0) out of memory
*** Check failure stack trace: ***
已放弃 (核心已转储)

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值