基于YOLOv3的猫狗分类遇到的问题

基于YOLOv3的猫狗分类遇到的问题

参考b站学习视频:
https://www.bilibili.com/video/BV1Hp4y1y788/?p=15&spm_id_from=pageDriver&vd_source=04e524adafca59c382b697248e46b881

下载了torch后,仍无法正常导入,是因为需要下载GPU版的torch

pip install torch===(要下的torch的版本)+cu110 -f https://download.pytorch.org/whl/torch_stable.html -i https://pypi.douban.com/simple/
(一般地,cu后面的数字,是根据cuda的版本定的,比如说,cuda版本为10.0,则为“cu100”,但是我cuda版本是100,直接用了110也行,看你报不报错吧)

train.py运行时遇到内存不足问题时

1、找到采样器中的__iter__方法,将n的参数设置小,最终规避报错。
参考路径:F:\Anaconda3\Conda\Lib\site-packages\torch\utils\data
在这里插入图片描述
2、修改num_workers为0,因为windows上无法使用线程
3、修改batch_size的值为1

train.py运行时遇到“ImportError: DLL load failed while importing _ufuncs: 找不到指定的程序”时

根据经验而言,如果你的python的版本是3.9及以上的话,建议直接换成3.7版本的
别问,问就是改了大半天,最后换成3.7版本的python就没有报错了(谁能懂我的痛)
后期回头补充:也可能是我的cuda版本和python版本不兼容的问题,具体大家可以参考下图
在这里插入图片描述

参考文章:
[1]: https://blog.csdn.net/zpc17875305019/article/details/113244333?spm=1001.2014.3001.5502
[2]: https://blog.csdn.net/weixin_43509698/article/details/113916524

  • 0
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值