Linux下VMamba 环境复现+环境测试

# 1. 创建自己的虚拟环境
conda create -n VMamba python=3.10.13
conda activate VMamba
# 2. cuda-11.8
conda install cudatoolkit==11.8 -c nvidia
# 3. torch torchvision torchaudio 与 官网命令一致
pip install torch==2.1.1 torchvision==0.16.1 torchaudio==2.1.1 --index-url https://download.pytorch.org/whl/cu118
# 4. nvcc
conda install -c "nvidia/label/cuda-11.8.0" cuda-nvcc
# 5. packaging
conda install packaging
# 6.openlab环境
pip install mmcv==2.1.0 -f https://download.openmmlab.com/mmcv/dist/cu118/torch2.1/index.html
pip install mmengine==0.10.1
pip install mmdet==3.3.0 mmsegmentation==1.2.2 mmpretrain==1.2.0
# 7. causal-conv1d== & mamba-ssm==1.1.1
pip install causal-conv1d==1.1.1
pip install mamba-ssm==1.1.2 #1.1.1实测会报错

causal-conv1d和mamba-ssm上述方法安装失败,可以见链接,采用whl安装。

关于NameError: name 'selective_scan_cuda_core问题,是跟cd kernels/selective_scan && pip install .这个安装不成功有关,(分析原因应该是cuda版本不对)

NameError: name 'selective_scan_cuda_core' is not defined. Did you mean: 'selective_scan_flop_jit'?

这边直接上我的环境信息吧,

certifi==2022.12.7
charset-normalizer==2.1.1
einops==0.8.0
filelock==3.13.1
fsspec==2024.2.0
fvcore==0.1.5.post20221221
huggingface-hub==0.23.0
idna==3.4
iopath==0.1.10
Jinja2==3.1.3
MarkupSafe==2.1.5
mpmath==1.3.0
networkx==3.2.1
ninja==1.11.1.1
numpy==1.26.3
packaging @ file:///home/conda/feedstock_root/build_artifacts/packaging_1637239678211/work
pillow==10.2.0
portalocker==2.8.2
pyparsing @ file:///home/conda/feedstock_root/build_artifacts/pyparsing_1652235407899/work
PyYAML==6.0.1
requests==2.28.1
safetensors==0.4.3
selective-scan @ file:///home/rycc/cc/VMamba/kernels/selective_scan
sympy==1.12
tabulate==0.9.0
termcolor==2.4.0
timm==0.9.16
torch==2.1.1+cu118
torchaudio==2.1.1+cu118
torchvision==0.16.1+cu118
tqdm==4.66.4
triton==2.1.0
typing_extensions==4.9.0
urllib3==1.26.13
yacs==0.1.8

测试代码,在classification/models/vmamba.py最后添加,并运行

device = torch.device("cuda:0")
hidden_dim = 3  # 假设 VSSBlock 的隐藏维度为 64
network = VSSM(hidden_dim).to('cuda:0')
# 随机生成一张输入图片作为示例
input_image = torch.randn(1, 3, 224, 224)  # 假设输入图片大小为 224x224,通道数为 3(RGB图像)
input_image = input_image.to(device)
# 前向传播
output = network(input_image)
# 输出结果的大小
print("Output shape:", output.shape)

输出结果,表示成功配置
在这里插入图片描述

训练命令 (目前数据集还未下载,后续更新)

python -m torch.distributed.launch --nnodes=1 --node_rank=0 --nproc_per_node=2 --master_addr="127.0.0.1" --master_port=29501  main.py  --cfg  /home/rycc/cc/VMamba/classification/configs/vssm/vmambav0_base_224.yaml

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

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

相关文章

苹果电脑怎么清内存?2024有哪些好用的工具?

在使用苹果电脑的过程中,我们可能会遇到系统运行缓慢、程序响应迟缓或频繁出现应用程序崩溃的情况,这些问题很可能是由于内存占用过高所导致。内存,或称为RAM(RandomAccessMemory),是计算机的临时存储区&am…

[C++][PCL]pcl安装包预编译包国内源下载地址

版本名称下载地址PCL-1.14.1-AllInOne-msvc2022-win64含pdb.zip点我下载PCL-1.14.0-AllInOne-msvc2022-win64含pdb.zip点我下载PCL-1.13.1-AllInOne-msvc2022-win64含pdb.zip点我下载PCL-1.13.0-AllInOne-msvc2022-win64含pdb.zip点我下载PCL-1.12.1-AllInOne-msvc2019-win64含…

超级好看的html网站维护源码

源码介绍 好看的html网站维护源码,源码由HTMLCSSJS组成,记事本打开源码文件可以进行内容文字之类的修改,双击html文件可以本地运行效果,也可以上传到服务器里面, 源码截图 源码下载 好看的html网站维护源码

py黑帽子学习笔记_网络编程工具

tcp客户端 socket.AF_INET表示使用标准IPV4地址和主机名 SOCK_STREAM表示这是一个TCP客户端 udp客户端 udp无需连接,因此不需要client.connect这种代码 socket.SOCK_DGRAM是udp的 tcp服务端 server.listen(5)表示设置最大连接数为5 发现kill server后端口仍占用…

【服务治理中间件】consul介绍和基本原理

目录 一、CAP定理 二、服务注册中心产品比较 三、Consul概述 3.1 什么是Consul 3.2 Consul架构 3.3 Consul的使用场景 3.4 Consul健康检查 四、部署consul集群 4.1 服务器部署规划 4.2 下载解压 4.3 启动consul 五、服务注册到consul 一、CAP定理 CAP定理&#xff…

i春秋-Backdoor

题目 考点 git源码泄露 Linux文件恢复 代码审计 http 解题 参考wp https://blog.csdn.net/cbhjerry/article/details/105791056https://www.pianshen.com/article/19461342501/扫描 题目给出提示:敏感文件泄漏 于是使用dirsearch扫一下 python dirsearch.py -…

【C++风云录】跨越时空的分析:古生物学与化石记录

古生物学数字化:C库的应用和影响 前言 在本文中,我们将深入探讨和评估几个重要的C库及其在古生物学和化石记录分析中的应用。此外,我们还将介绍MorphoSource API,以及使用C进行API接入的过程。文章还会讲述化石校准工具的重要性…

STM32H5 擦除flash时,GetSector进入Error_Handler

背景及问题:开发项目时,操作内部flash,调用getsetcor时,进入hardfault,经调试发现FLASH_BANK_SIZE读不到值,FLASH_BANK_SIZE取值来源于以下步骤:一时没有任何思路 ((((*((uint16_t *)FLASHSIZE…

ICode国际青少年编程竞赛- Python-4级训练场-while语句综合

ICode国际青少年编程竞赛- Python-4级训练场-while语句综合 1、 for i in range(4):while not Flyer[i].disappear():wait()Spaceship.step(6)Spaceship.turnLeft()2、 Dev.turnLeft() for i in range(4):Spaceship.step(2)while Flyer[i].disappear():wait()Dev.step(4)Dev.…

研究生写论文常用工具以及如何快速阅读外文

研究生在写论文时经常需要用到一些常用的科研工具,这里介绍8款,可以根据需求使用: 1、文献检索工具:如Google Scholar、Web of Science、PubMed等,用于查找相关文献和资料。 2、文献管理工具:如EndNote、…

Failed to parse source map (@toast-ui/editor/dist/purify.js.map)

使用 toast-ui-editor 时出现报错:Failed to parse source map (toast-ui/editor/dist/purify.js.map) 解决方法很简单: "start": "set "GENERATE_SOURCEMAPfalse" && react-scripts start ",在启动脚本时添加执…

程序员代码面试指南题目解析(一)

题目一:如何仅用递归函数和栈操作逆序一个栈 题目要求: 一个栈依次压入 1、2、3、4、5,那么从栈顶到栈底分别为5、4、3、2、1。将这个栈 转置后,从栈顶到栈底为 1、2、3、4、5,也就是实现栈中元素的逆序,但…

【吴恩达机器学习-week2】多个变量的线性回归问题

文章目录 1.1 目标2 问题陈述2.1 包含我们示例的矩阵 X2.2 参数向量 w \mathbf{w} w 和 b b b 3 使用多个变量进行模型预测3.1 逐元素单独预测3.2 单一预测,向量 4 使用多个变量计算成本5 使用多个变量的梯度下降5.1 使用多个变量计算梯度 多个变量的梯度下降小结…

软件测试概述

目录 引言 1. 软件测试的原因 2. 软件缺陷的由来 3. 软件测试的发展历程 4. 软件测试的定义 5. 软件测试的层次和类型 6. 软件测试的过程 结语 引言 在当今以软件为中心的时代,软件已无处不在,深刻影响着我们的生活、工作和娱乐方式。从智能手机…

Vue 3 中使用 const model=defineModel() 简化 v-model 的技巧

当我们在 Vue 3 中处理表单输入时,经常会使用 v-model 来实现数据的双向绑定。然而,在某些情况下,我们可能需要对 v-model 进行一些额外的处理,例如对输入值进行格式化、验证或者其他操作。这时,可以使用 const model …

搜歌网搜索各种类型音乐,统统歌曲转换格式mp3,轻松实现音乐自由!

在互联网的广阔天地中,音乐爱好者们总能找到满足自己需求的平台。其中,支持全网搜歌的网站无疑是一个值得推荐的音乐探索乐园。无论是寻找经典老歌,还是发掘新兴音乐,搜他们都能为音乐爱好者提供一站式的服务。 一般支持全网搜索…

值得收藏!!《软考信息处理技术员》必背100母题,轻松45+

距离软考考试的时间越来越近了,趁着这两周赶紧准备起来 今天给大家整理了——软考信息处理技术员100道经典母题,年年从里面抽,有PDF,可打印,每天刷几道。 第一章 电脑的基本操作 1、( )不是国产…

Linux线程(二)线程互斥

目录 一、为什么需要线程互斥 二、线程互斥的必要性 三、票务问题举例(多个线程并发的操作共享变量引发问题) 四、互斥锁的用法 1.互斥锁的原理 2、互斥锁的使用 1、初始化互斥锁 2、加锁和解锁 3、销毁互斥锁(动态分配时需要&#…

在AWS EKS中使用Socket.IO

背景 需要在K8S集群中提供websocket服务。后端使用的是netty-socketio库实现了Socket.IO方式的ws服务。前端直接使用Socket.IO 客户端js实现。这里重点介绍AWS EKS中怎么通过启用ALB粘性会话的方式支持ws服务。 svc.yaml apiVersion: v1 kind: Service metadata:labels:app:…

RFID在汽车制造中的应用如何改变行业

随着工业4.0和中国制造2025的推进,企业对于智能化、自动化的需求日益增长,RFID射频技术在制造业中已经相当普遍了。在如今这瞬息万变的行业与时代中,RFID技术可以帮助企业获得竞争优势,简化日益复杂的生产流程,推动企业…