前言

最近在使用Vision Mamba需要用到这两个库,遇到一些问题,将问题和解决办法记录一下

causal_conv1d 和 mamba_ssm 安装方法

库的地址:https://pypi.org/project/mamba-ssm/1.1.4/

# 官方给的(powershell无法使用这个版本限定的格式,不推荐)
pip install causal-conv1d>=1.1.0,<1.2.0
# 推荐使用这行命令
pip install causal-conv1d==1.1.1
# 这里使用的是1.1.4
pip install mamba-ssm==1.1.4

causal_conv1d安装问题

  1. 报错中出现没有识别到 nvcc 的问题
    需要下载 CUDA Toolkit 并进行安装(因为安装pytorch时会自动安装一个cuda所以很有可能没有单独装一个,尽量安装跟conda环境相同版本的)

  2. error: Microsoft Visual C++ 14.0 or greater is required.
    完整报错:

error: Microsoft Visual C++ 14.0 or greater is required. Get it with "Microsoft C++ Build Tools": https://visualstudio.microsoft.com/visual-cpp-build-tools/
      [end of output]

解决办法一:按照这个流程去安装
解决办法二:如果不想去安装VS的话,可以下载这个wheel,本地安装参考链接
whl下载地址

Logo

旨在为数千万中国开发者提供一个无缝且高效的云端环境,以支持学习、使用和贡献开源项目。

更多推荐