当前位置:   article > 正文

causal_conv1d和mamba_ssm安装报错问题_causal-conv1d

causal-conv1d

在安装mamba环境时要用到 causal_conv1d包和mamba_ssm

  1. pip install causal-conv1d
  2. pip install mamba_ssm

我在安装时会报以下图片的错误

看到报错表示CUDA版本要高于 11.6就去查看nvidia-smi发现cuda版本是11.7超过了11.6但仍然报错。

然后我看报错提到了nvcc,又从理清GPU、CUDA、CUDA Toolkit、cuDNN关系以及下载安装_cudatoolkit-CSDN博客了解到cuda会有两种版本,nvidia-smi是 CUDA Driver API(驱动API)的版本,也是当前驱动支持的最高CUDA版本,如在这里是11.7。而nvcc的结果是对应 CUDA Runtime API(运行API)的版本:如这里的nvcc的版本是10.0

另一台linux之前一下安装就成功了发现他的nvcc的cuda版本是11.7,查看报错和资料了解到,可能是nvcc的原因,需要去安装cudatoolkitcausal_conv1d和mamba_ssm安装问题_windows安装causal-conv1d-CSDN博客

  1. #安装cudatoolkit
  2. conda install -c conda-forge cudatoolkit-dev

此时nvcc的cuda版本升级到了11.7

这时就可安装causal-con1d了

成功安装 接下来就可安装mamba_ssm

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/菜鸟追梦旅行/article/detail/565393
推荐阅读
相关标签
  

闽ICP备14008679号