Mobilemamba

Question

1、问题现象描述：
mobilemamba的环境如何配置啊，自己配环境配了好久根本弄不好，有人给我说这个要依赖apex什么的，有成功配好mobilemamba环境的嘛，求指点！
2、问题相关代码运行结果，报错内容：
①这是GitHub上面官方给出的配置环境命令：

pip3 install torch==2.1.2 torchvision==0.16.2 torchaudio==2.1.2 --index-url https://download.pytorch.org/whl/cu118
pip3 install timm==0.9.16 tensorboardX einops torchprofile fvcore==0.1.5.post20221221 triton==2.1.0
cd model/lib_mamba/kernels/selective_scan && pip install . && cd ../../../..
git clone https://github.com/NVIDIA/apex && cd apex && pip3 install -v --disable-pip-version-check --no-cache-dir --global-option="--cpp_ext" --global-option="--cuda_ext" ./ (optional)

②我在运行完第三个命令后出现报错：

3、我的初步解决思路：
①首先我以为是系统中CUDA配置出现问题，进行了如下代码调试：

(mobilemamba) [test04@fm1-login1 ~]$ python
Python 3.10.16 (main, Dec 11 2024, 16:24:50) [GCC 11.2.0] on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> import torch
>>> print(torch.cuda.is_available())
False
>>> print(torch.version.cuda)
12.1

这里显示我的cuda是false，应该是我用的是服务器，还没有申请GPU，所以显示false。并且我的cuda版本显示是12.1，而前面第一条命令要求的是11.8，所以我也不清楚我的报错跟这个有没有关系。
②然后我跑去咸鱼问了一下，他说这个需要apex，得以来Linux，我不太了解apex是什么，也不清楚他说的对不对

4、我创建的虚拟环境是python=3.10，cuda版本是12.1

请大家帮帮我，这个环境搭建问题困扰我很久很久了，不知道为什么mobilemamba这个网络的环境搭建这么麻烦，谢谢大家！
项目代码：https://github.com/lewandofskee/MobileMamba

Answer 1

阿里嘎多学长整理AIGC生成，因移动端显示问题导致当前答案未能完全显示，请使用PC端查看更加详细的解答过程

MobileMamba环境配置

你好，MobileMamba是一个基于PyTorch的目标检测框架，配置环境需要一些步骤。下面是一个简单的配置步骤：

1. 安装PyTorch和相关依赖项

pip install torch torchvision

1. 安装MobileMamba

pip install mobilemamba

1. 安装apex（optional）

pip install apex

1. 配置环境变量

import os
os.environ["CUDA_VISIBLE_DEVICES"] = "0"  # 设置GPU设备
os.environ["MOBILEMAMBA_HOME"] = "/path/to/mobilemamba"  # 设置MobileMamba的安装目录

1. 验证环境

import mobilemamba
print(mobilemamba.__version__)

如果你已经安装了PyTorch和MobileMamba，但仍然无法正确配置环境，可以尝试以下步骤：

• 检查PyTorch和MobileMamba的版本是否正确
• 检查apex是否正确安装
• 检查环境变量是否正确设置
• 尝试使用mobilemamba命令来验证环境

如果你仍然无法解决问题，可以提供更多的错误信息和环境配置信息，我将尽力帮助你解决问题。