当前位置: 首页 > news >正文

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

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

基础配置:
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.rkmt.cn/news/57276.html

相关文章:

  • CMake构建学习笔记30-Ceres Solver库的构建
  • curl 命令使用笔记
  • 2025年口碑好的电厂清淤机器人厂家最新用户好评榜
  • 2025年靠谱的极低压抗污染反渗透膜厂家最新TOP排行榜
  • 2025年11月留学生回国求职机构推荐榜单:权威机构列表与选择指南
  • 2025年11月留学生回国求职机构推荐:五大权威机构榜单与选择指南
  • 2025年评价高的快速离心浓缩干燥器TOP品牌厂家排行榜
  • 2025年比较好的组合式恒温 振荡培养箱最新TOP品牌厂家排行
  • 2025年专业的输送机高评价厂家推荐榜
  • 2025年诚信的镁制硅晶防火风管厂家推荐及选购指南
  • 2025年靠谱的试验机行业内知名厂家排行榜
  • 2025年11月留学生找实习机构推荐:权威市场报告解读与口碑机构列表
  • 2025年质量好的自动叠皮机厂家推荐及采购指南
  • 2025年耐用的钢板预处理线TOP品牌厂家排行榜
  • 2025年口碑好的F40喷涂四氟厂家最新TOP排行榜
  • 2025年评价高的鲜面条生产线厂家最新TOP排行榜
  • 2025年比较好的佛山负氧离子床垫行业内口碑厂家排行榜
  • 聚焦2025:食品冻干机厂家推荐大比拼,果蔬冷冻干燥机推荐
  • 2025年11月美国求职机构推荐:权威榜单与深度选择指南
  • 2025年11月美国求职机构推荐榜单及选择指南:基于多维度对比的权威解析
  • 2025年优秀的养殖专用防渗土工膜工程首选品牌推荐榜
  • 2025年靠谱的医养家具厂家最新推荐排行榜
  • 2025年11月抛丸机厂家权威推荐列表:用户评价与服务质量全视角
  • 2025年有实力的日光温室大棚最新TOP品牌厂家排行
  • 漏洞挖掘之旅:一位漏洞猎人的网络安全征程
  • 2025年正规的电缆桥架厂家最新用户好评榜
  • 深入解析:C++:从0开始学习链表(练习)
  • 2025年耐用的弯管加工厂家最新热销排行
  • OOP-实验3
  • 2025年优秀的火山岩滤料行业内知名厂家排行榜