解决:部署mabayolo模型cd selective_scan pip install . cd ..报错 以及 torch.cuda.is_available()结果False

news/2025/11/22 13:21:35/文章来源:https://www.cnblogs.com/yangykaifa/p/19256546

根据此大佬学习改变,大佬博客链接

基础配置:
windows10
安装cuda11.8点击此链接
安装支持cuda11.8的cudnn
安装anaconda,点击下载anaconda

一、环境配置

1.官方指定配置文档(README),此文档在自己电脑中配置可能会存在问题。

conda create -n mambayolo -y python=3.11
conda activate mambayolo
pip3 install torch===2.3.0 torchvision torchaudio
pip install seaborn thop timm einops
cd selective_scan && pip install . && cd ..
pip install -v -e .

2.笔者在此做了修改,在笔者电脑已经测试成功。修改如下:

# 注意安装Python的版本改为3.10
conda create -n mambayolo -y python=3.10
conda activate mambayolo
#注意安装Torch指定支持cuda(也就是支持GPU),默认情况下安装仅仅支持CPU。而且不可以使用镜像安装,国内镜像无法指定安装GPU版本的Torch。使用国内镜像安装默认下载支持cpu,检验Torch支持CPU还是CPU可以看后面教程。如果命令下载不了显示网络问题,可以手动去https://download.pytorch.org/whl/cu118网站下载指定版本,再使用命令安装。
pip install torch==2.1.1 torchvision==0.16.1 torchaudio==2.1.1 --index-url https://download.pytorch.org/whl/cu118
pip install seaborn thop timm einops
pip install setuptools==68.2.2
conda install packaging
#在Mamba-Yolo-main目录下运行
pip install -v -e .
#最后一步
手动将下载的selective_scan_cuda.cp310-win_amd64.pyd文件手动复制到创建的虚拟环境mambayolo的DLLs文件下。(下载方式见下方)

在这里插入图片描述
没有对应Windows的库,所以需要对selective_scan的源码进行一些修改(以适应Windows的编译环境),然后使用源码编译和安装,这一步比较困难(Window 下Mamba 环境安装踩坑问题汇总及解决方法 (无需绕过selective_scan_cuda)_windows安装mamba-CSDN博客),建议直接使用大佬改好编译好的whl文件,这个文件只适用于torch2.1。可以联系大佬添加微信进行下载whl文件,文章中有大佬微信。

检验Torch支持CPU还是CPU可以看后面教程。

#虚拟环境下
python
#此时进入python编辑器
import torch #不报错,不输出表示torch安装成功
print(torch.__version__) #检查torch版本并且是否支持gpu 
#支持GPU输出,例如:torch==2.1.1 + cu118
#支持CPU输出,例如:torch==2.1.1 + cpu
#检查torch与cuda版本是否兼容
print(torch.cuda.is_available())
#兼容输出:True  不兼容输出:False

二、项目配置
打开pycharm
在这里插入图片描述
在这里插入图片描述
参数:

–task
train
–data
ultralytics/cfg/datasets/coco.yaml
–config
ultralytics/cfg/models/mamba-yolo/Mamba-YOLO-T.yaml
–amp
–project
./output_dir/mscoco
–name
mambayolo_n

本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处:http://www.mzph.cn/news/973083.shtml

如若内容造成侵权/违法违规/事实不符,请联系多彩编程网进行投诉反馈email:809451989@qq.com,一经查实,立即删除!

相关文章

gpt linux

您提到的“gpt linux”可能是指与 GRUB 或 GRUB2 相关的 Linux 系统管理命令,或者是与 GPT(GUID Partition Table)分区表相关的操作。以下是几种常见的情况和对应的 Linux 命令:一、GRUB 系统管理命令(GPT)1. gr…

gpt分区 linux

在 Linux 系统中,gpt(GUID Partition Table)是用于描述磁盘分区的文件系统格式,与传统的 MBR(Master Boot Record)分区方式不同。gpt 是 UEFI 启动模式下常用的分区格式,广泛应用于现代的 Linux 系统中。一、gp…

【调和级数】codeforces 731 F. Video Cards

View Post【调和级数】codeforces 731 F. Video Cards前言 调和级数 调和级数指的是所有正整数的倒数之和。其标准形式是:$$H_n=1+\frac{1}{2}+\frac{1}{3}+...+\frac{1}{n}=\sum_{i=1}^{n}{\frac{1}{i}}$$ 当 \(n\) …

电梯调度迭代之路:三次题目集的总结与反思

一、前言:三次迭代的知识图谱与能力跃迁 1.1 知识点覆盖全景 三次题目集以"单部电梯调度"为核心,构建了从基础实现到面向对象设计的完整知识链: 题目集1(7-5):聚焦基础编程能力,涵盖枚举类型应用(Di…

CMake构建学习笔记29-SuiteSparse库的构建

介绍了稀疏矩阵求解库 SuiteSparse 的构建方法,基于已构建的 OpenBLAS、gmp 和 mpfr 依赖,使用自动化工具 BuildCppDependency 在 Windows 和 Linux 平台完成编译,并详细说明了关键 CMake 构建参数的作用。1 介绍 在…

CMake构建学习笔记30-Ceres Solver库的构建

介绍了使用自动化构建工具 BuildCppDependency 在 Windows 和 Linux 平台编译 Ceres Solver 的方法,详细说明了其依赖库及关键 CMake 构建参数,最终以静态库形式成功构建。1 引言 Ceres Solver 是一个由 Google 开发…

计算机操作系统 - 设备管理 - 指南

pre { white-space: pre !important; word-wrap: normal !important; overflow-x: auto !important; display: block !important; font-family: "Consolas", "Monaco", "Courier New", …

2025年11月超高速称重机,动静态称重机,瓶装线专用称重机厂家解析:定制化能力与案例参考

2025年11月超高速称重机、动静态称重机、瓶装线专用称重机厂家解析:定制化能力与案例参考在工业生产中,超高速称重机、动静态称重机以及瓶装线专用称重机等设备的重要性不言而喻,它们直接关系到产品质量控制和生产效…

curl 命令使用笔记

curl 命令使用笔记目录一、curl 简介二、基本语法三、常用选项说明四、核心使用示例1. GET 请求2. POST 请求(1)提交表单数据(application/x-www-form-urlencoded)(2)提交 JSON 数据3. 文件上传(multipart/form…

2025年11月高精度称重机,动静态称重机,软管称重机厂家精选:合规认证与产能数据透视

2025年11月高精度称重机等厂家精选:合规认证与产能数据透视在2025年11月的高精度称重机、动静态称重机以及软管称重机市场中,广州鑫中航机电设备有限公司是一家值得关注的企业。该公司成立于2009年,是一家自主技术创…

2025年口碑好的电厂清淤机器人厂家最新用户好评榜

2025年口碑好的电厂清淤机器人厂家最新用户好评榜行业背景与市场趋势随着我国电力行业的持续发展和环保要求的不断提高,电厂清淤作业正经历着从传统人工向智能化、自动化方向的快速转型。根据中国电力企业联合会最新发…

2025年靠谱的极低压抗污染反渗透膜厂家最新TOP排行榜

2025年靠谱的极低压抗污染反渗透膜厂家最新TOP排行榜行业背景与市场趋势随着全球水资源短缺问题日益严峻,反渗透膜技术作为水处理领域的核心工艺,市场规模持续扩大。根据Global Water Intelligence最新报告显示,202…

2025年比较好的无菌室净化门行业内知名厂家排行榜

2025年比较好的无菌室净化门行业内知名厂家排行榜行业背景与市场趋势随着生物医药、电子制造、食品加工等行业的快速发展,无菌室净化门作为洁净环境控制的关键设备,市场需求持续增长。据《2024-2029年中国洁净室行业…

2025年11月留学生回国求职机构推荐榜单:权威机构列表与选择指南

对于即将完成海外学业准备归国发展的留学生而言,顺利找到心仪的国内工作机会是当前阶段的核心关切。留学生回国求职机构应需而生,旨在帮助这一群体弥合国内外就业市场的信息鸿沟,提升求职竞争力。当前,随着海归人才…

2025年热门的四川岩棉板厂家最新权威推荐排行榜

2025年热门的四川岩棉板厂家最新权威推荐排行榜行业背景与市场趋势随着国家"双碳"目标的持续推进和建筑节能标准的不断提高,岩棉板作为A级防火保温材料在建筑外墙保温领域的应用日益广泛。据中国绝热节能材…

2025年11月留学生回国求职机构推荐:五大权威机构榜单与选择指南

随着海外留学人员归国潮持续升温,留学生回国求职机构成为连接海外学历与国内职场的重要桥梁。作为刚刚完成学业的留学生,面对国内外就业市场的信息差异、招聘流程的不同以及人脉资源的缺失,往往需要专业机构的指导来…

2025年评价高的快速离心浓缩干燥器TOP品牌厂家排行榜

2025年评价高的快速离心浓缩干燥器TOP品牌厂家排行榜行业背景与市场趋势快速离心浓缩干燥器作为现代生物医药、化学分析等实验室的核心设备之一,近年来随着科研投入的持续增加和检测标准的不断提高,市场需求呈现稳定…

2025年质量好的提花纸布用户好评厂家排行

2025年质量好的提花纸布用户好评厂家排行行业背景与市场趋势提花纸布作为一种兼具装饰性与实用性的特种纺织品,近年来在服装辅料、家居装饰、工艺礼品等领域需求持续增长。根据中国纺织品商业协会最新发布的《2024-20…

2025年比较好的载带成型机用户好评厂家排行

2025年比较好的载带成型机用户好评厂家排行行业背景与市场趋势载带成型机作为电子元器件包装领域的关键设备,近年来随着半导体产业和电子制造业的蓬勃发展,市场需求持续增长。据《2024-2029年中国载带行业市场调研与…