赞
踩
在安装mamba环境时要用到 causal_conv1d包和mamba_ssm
- pip install causal-conv1d
- 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博客
- #安装cudatoolkit
- conda install -c conda-forge cudatoolkit-dev
此时nvcc的cuda版本升级到了11.7
这时就可安装causal-con1d了
成功安装 接下来就可安装mamba_ssm
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。