主要是因为torch版本的问题,改代码太麻烦了(我也不会)
解决办法:重新搭建环境使torch版本低于1.10.0
比如重新搭建环境名为yolo:conda create -n yolo5 python==3.8.5
然后激活环境:conda activate yolo
再安装torch:conda install pytorch==1.8.0 torchvision torchaudio cudatoolkit=10.2
欢迎分享,转载请注明来源:内存溢出
主要是因为torch版本的问题,改代码太麻烦了(我也不会)
解决办法:重新搭建环境使torch版本低于1.10.0
比如重新搭建环境名为yolo:conda create -n yolo5 python==3.8.5
然后激活环境:conda activate yolo
再安装torch:conda install pytorch==1.8.0 torchvision torchaudio cudatoolkit=10.2
欢迎分享,转载请注明来源:内存溢出
评论列表(0条)