yolov5 本地训练
YOLOv5 | Kaggle
直接gitclone他的源码用Vscode看(也可以直接把jupyter下下来)
他要1.8,我的是2.7,他这个代码可能有点年头了
两年前了
他的环境
我的环境
我就是不懂为什么清华源的torch windows默认下出来是cpu版本
.
在终端里重新下了一遍
要4g显存,我这电脑3g emm
torch+cu版本下高了又要重下
没高啊
用windows跑就是错误
参考如何解决安装的 CUDA 版本和 PyTorch 版本的兼容性问题-2 - 技术栈
jupyter pip install看不到过程
windows给我的就是这种感觉,我电脑还慢
torchvison要一起下,不然之后下torchvison版本太高会把当前的torch删掉
之前不小心吧环境变量删了
重装git它会往环境变量里面加这个
8g内存跑不了,它内存29g
参考:OSError: [WinError 1455] 页面文件太小,无法完成操作。 Error loading “C:\Users\hsadkhaida\AppData\Roaming\Python\Py-CSDN博客
我看它峰值也才5g啊
给它加1t
训练到一半爆掉了
设置移动硬盘不行
内存满了,gpu使用为0,换笔记本跑看看
8g内存3g显存试跑成功,就是很慢
给多一点看看
代码移动到固态看看
火狐浏览器占用内存500m,vscode占用1g
笔记本也才需要32s
设置虚拟内存移动到固态后跑完一趟要10分钟,快了一点点 ,内存会在90%左右
好像不吃gpu