基于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