pytorch常见问题及解决方案

1、pycharm终端禁止运行脚本

解决方案:

  • “以管理员身份运行”Windows PowerShell(开始菜单搜索),出现与上图一样的错误提示;

  • 输入”get-ExecutionPolicy“,输出”Restricted“,即脚本执行策略受限。

  • 更换脚本执行策略:”set-ExecutionPolicy RemoteSigned“,然后输入”Y“

  • 更换完成后,再次使用命令”get-ExecutionPolicy“,查看脚本执行策略。问题解决。

2、开源项目YOLOv5 5.0软件包安装失败

解决方案:

  • 进入anaconda官网,搜索”opencv-python”,如图,点击第一条进去,复制下载命令,去pycharm终端pytorch环境下,运行下载安装。

  • 或者在pycharm终端输入“pip install oprncv-python”回车执行安装。

  • 在pycharm终端输入“pip install thop”回车执行安装。

  • 进入anaconda官网,搜索”pycocotools”,如图,点击第一条进去,复制下载命令(“conda install -c conda-forge pycocotools”),去pycharm终端pytorch环境下,运行下载安装。

3、运行YOLOv5 detect.py运行错误

报错1:AttributeError: ‘Upsample’ object has no attribute ‘recompute_scale_factor’

(14条消息) YOLOV5 | AttributeError: ‘Upsample‘ object has no attribute ‘recompute_scale_factor‘ 问题解决 亲测有效_RodgeH的博客-优快云博客

解决方案:

  • 进入倒数第二个报错行路径,进入编辑页

  • 将原代码(156-157行)修改为如下所示(158行):

保存后再次运行detect.py,即可。

报错2:UserWarning: torch.meshgrid: in an upcoming release, it will be required to pass the inddexing argument

(14条消息) Yolov5训练时报错:UserWarning: torch.meshgrid: in an upcoming release, it will be required to pass the ind_奋斗的豆腐的博客-优快云博客

根据报错提示,进入错误路径,将其修改为

保存,运行成功。

报错3:cuda爆显存,out of memory

解决方案:

  • 在main函数中,batch_size部分,调低batch_size值,默认为16,可以适当调低

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值