首页 > 其他分享 >关于Vmamba模块安装SSM包环境问题

关于Vmamba模块安装SSM包环境问题

时间:2024-07-09 16:26:50浏览次数:13  
标签:Vmamba echo SSM CUDA 模块 HOME 安装 cuda

提示:文章写完后,目录可以自动生成,如何生成可参考右边的帮助文档

@TOC关于Vmamba模块安装SSM包环境问题


前言

提示:这里可以添加本文要记录的大概内容:

最近在服务器上安装VMamba环境,记录一下解决问题的方法。

因为网上租的服务器他的CUDA不在系统的环境变量当中,所以,下载了causal-conv1dmamba-1p1p1cd分别进入这两个包:

pip install -e .

安装这个两个包时,会报错。

解决方法

添加环境变量
先查看有没有:

echo $CUDA_HOME

刚配好的conda环境中即使安装了有关于cuda的包,这个大概率是没有的。

export CUDA_HOME=/usr/local/cuda
source ~/.bashrc

设置好后,查看

echo $LD_LIBRARY_PATH

如果有显示就成功了
然后再安装这两个包就很顺利了。注意:前提是已经完成了网上大多数教程的安装后,例如CUDA版本在11.8以上等等问题就不赘述了

标签:Vmamba,echo,SSM,CUDA,模块,HOME,安装,cuda
From: https://blog.csdn.net/xiequan1/article/details/140277777

相关文章