win10 安装Mish-CUDA 遇到问题解决

在安装的时候遇到了很多问题。

首先就是ninja报错Command '['ninja', '-v']' returned non-zero exit status 1.

然后是gcc问题

解决方法:我怀疑是torch版本不对,我使用了pytorch1.7.0和1.8.0;

所以开始更新cuda+cudnn=11.3;pytorch=1.10

此时安装成功,因为跳过了ninja的代码,

然后对mish-cuda进行,python setup.py build 和 install就可以了

ps:我修改了mish-cuda的一些代码,如果有小伙伴不能安装的话可以私我,或者在我发布的资源里找。

  • 1
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
需要学习ubuntu系统上YOLOv4的同学请前往:《YOLOv4目标检测实战:原理与源码解析》 【为什么要学习这门课】 Linux创始人Linus Torvalds有一句名言:Talk is cheap. Show me the code. 冗谈不够,放码过来! 代码阅读是从基础到提高的必由之路。尤其对深度学习,许多框架隐藏了神经网络底层的实现,只能在上层调包使用,对其内部原理很难认识清晰,不利于进一步优化和创新。YOLOv4是最近推出的基于深度学习的端到端实时目标检测方法。YOLOv4的实现darknet是使用C语言开发的轻型开源深度学习框架,依赖少,可移植性好,可以作为很好的代码阅读案例,让我们深入探究其实现原理。【课程内容与收获】 本课程将解析YOLOv4的实现原理和源码,具体内容包括:- YOLOv4目标检测原理- 神经网络及darknet的C语言实现,尤其是反向传播的梯度求解和误差计算- 代码阅读工具及方法- 深度学习计算的利器:BLAS和GEMM- GPU的CUDA编程方法及在darknet的应用- YOLOv4的程序流程- YOLOv4各层及关键技术的源码解析本课程将提供注释后的darknet的源码程序文件。【相关课程】 除本课程《Windows版YOLOv4目标检测:原理与源码解析》外,本人推出了有关YOLOv4目标检测的系列课程,包括:《Windows版YOLOv4目标检测实战:训练自己的数据集》《Windows版YOLOv4-Tiny目标检测实战:训练自己的数据集》《Windows版YOLOv4目标检测实战:人脸口罩佩戴检测》《Windows版YOLOv4目标检测实战:中国交通标志识别》建议先学习一门YOLOv4实战课程,对YOLOv4的使用方法了解以后再学习本课程。【YOLOv4网络模型架构图】 下图由白勇老师绘制  
根据引用\[1\]和引用\[2\]的内容,如果在训练过程中遇到了ModuleNotFoundError: No module named ‘mish_cuda’的错误,可以尝试根据错误提示将所有的from mish_cuda import MishCuda as Mish替换为其他代码。具体来说,可以将以下代码替换为原来的代码: ```python from mish_cuda import MishCuda as Mish ``` 替换为: ```python class Mish(nn.Module): def __init__(self): super().__init__() def forward(self, x): x = x * (torch.tanh(F.softplus(x))) return x ``` 此外,根据引用\[3\]的内容,如果你想安装mish-cuda,可以按照以下步骤进行操作: 1. 克隆yolov4的代码库:git clone https://github.com/WongKinYiu/PyTorch_YOLOv4.git 2. 准备数据集文件夹 3. 根据需要修改配置文件和网络结构 4. 配置weights文件 5. 进行训练 希望这些信息对你有帮助! #### 引用[.reference_title] - *1* [YOLO-V4 Pytorch版本训练自建数据集和预测](https://blog.csdn.net/worewolf/article/details/117425497)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v91^insert_down1,239^v3^insert_chatgpt"}} ] [.reference_item] - *2* *3* [【YOLO】pytorch版yolov4训练自己的数据集](https://blog.csdn.net/wxd1233/article/details/117781437)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v91^insert_down1,239^v3^insert_chatgpt"}} ] [.reference_item] [ .reference_list ]
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值