自定义博客皮肤VIP专享

*博客头图：

点击选择上传的图片

格式为PNG、JPG，宽度*高度大于1920*100像素，不超过2MB，主视觉建议放在右侧，请参照线上博客头图

请上传大于1920*100像素的图片！

博客底图：

点击选择上传的图片

图片格式为PNG、JPG，不超过1MB，可上下左右平铺至整个背景

栏目图：

点击选择上传的图片

图片格式为PNG、JPG，图片宽度*高度为300*38像素，不超过0.5MB

主标题颜色：

RGB颜色，例如：#AFAFAF

Hover：

RGB颜色，例如：#AFAFAF

副标题颜色：

RGB颜色，例如：#AFAFAF

预览取消提交

自定义博客皮肤

上一步保存

IT_AGENT_PY的博客

Alex landi

博客等级

码龄7年

29
原创

83
点赞

217
收藏

33
粉丝

关注

私信

分类专栏

最新评论

【MambaSR复现】【Windows系统下Mamba环境配置】triton 、causal conv1d和mamba_ssm模块配置保姆教程
周时恒: (py38) PS D:\anaconda\envs\py38\Scripts\causal-conv1d> pip install . Processing d:\anaconda\envs\py38\scripts\causal-conv1d Preparing metadata (setup.py) ... error error: subprocess-exited-with-error × python setup.py egg_info did not run successfully. │ exit code: 1 ╰─> [6 lines of output] Traceback (most recent call last): File "<string>", line 2, in <module> File "<pip-setuptools-caller>", line 34, in <module> File "D:\anaconda\envs\py38\Scripts\causal-conv1d\setup.py", line 21, in <module> from torch.utils.cpp_extension import BuildExtension, CppExtension, CUDAExtension, CUDA_HOME, HIP_HOME ImportError: cannot import name 'HIP_HOME' from 'torch.utils.cpp_extension' (D:\anaconda\envs\py38\lib\site-packages\torch\utils\cpp_extension.py) [end of output] note: This error originates from a subprocess, and is likely not a problem with pip. error: metadata-generation-failed 失败了，这怎么解决啊
【MambaSR复现】【Mamba环境配置】No Module named “causal_conv1d_cuda”、“selective_scan_cuda”报错解决
Catnip807: 您好，我这里使用了ubuntu2204还是会碰到一样的问题，请问您是怎么解决的呢？
【MambaSR复现】【Mamba环境配置】No Module named “causal_conv1d_cuda”、“selective_scan_cuda”报错解决
Alex landi: 哥们，搞了好久终于搞明白了，mamba就不是给windows玩的，换linux跑吧，linux下两小时就跑通了windows下casual-conv1d和mamba-ssm两个包会出很多问题，一个bug接一个，我都不知道解决了多少个bug最后一样的出现训练时候会发现GPU占满，根本跑不了
【MambaSR复现】【Mamba环境配置】No Module named “causal_conv1d_cuda”、“selective_scan_cuda”报错解决
Alex landi: 我也遇到这个问题了，你有解决吗？
【MambaSR复现】【Mamba环境配置】No Module named “causal_conv1d_cuda”、“selective_scan_cuda”报错解决
Jwzhang7801: 改了后没有报错，但是显存24G占满，训练开始之后等很久一个epoch没跑完