[最佳实践] conda环境内安装cuda 和 Mamba的安装

Mamba安装失败的过程中,causal-conv1d安装报错为连接超时

key word: vision mamba, DL ,深度学习 ,mamba unet,mamba环境安装

Mamba安装

主要故障是 pip install causal-conv1d1.2.0和 pip install mamba-ssm1.2.0 安装失败

安装时间比较长,请耐心等待

解决方案

解决方法位置
受到启发运行Mamba项目时无法直接用pip install安装causal_conv1d和mamba_ssm_pip install causal-conv1d编译文件-CSDN博客

本地安装causal-conv1d时,一定要检查机器的gcc和g++版本,本人默认是gcc5就会编译报错,gcc9就能安装成功
gcc -V 可查看Gcc版本
安装时间比较长,请耐心等待
Ubuntu下gcc多版本共存和版本切换_ykrgcc-CSDN博客这里详细讲述了gcc版本切换

conda create -n your_env_name python=3.10.13
conda activate your_env_name
conda install cudatoolkit==11.8 -c nvidia
pip install torch==2.1.1 torchvision==0.16.1 torchaudio==2.1.1 --index-url https://download.pytorch.org/whl/cu118
conda install -c "nvidia/label/cuda-11.8.0" cuda-nvcc
conda install packaging
git clone https://github.com/Dao-AILab/causal-conv1d.git 
cd causal-conv1d 
git checkout v1.2.0 # current latest version tag 
CAUSAL_CONV1D_FORCE_BUILD=TRUE pip install .
git clone https://github.com/state-spaces/mamba.git
cd ../mamba
git checkout v1.2.0 # current latest version tag
MAMBA_FORCE_BUILD=TRUE pip install .

成功预览

image.png
下面是个人的安装尝试,均以失败告终。

失败经历

经历一

:::info

  1. Ubuntu内部先安装cuda11.8和cudnn
  2. 然后安装pytorch
  3. 然后安装pip install causal-conv1d==1.2.0 ,然后就报错了。都没等到安装Manba
    :::

经历二

然后通过观察
Mamba 环境安装踩坑问题汇总及解决方法_building wheel for causal-conv1d (setup.py) …-CSDN博客
调整为
:::info

  1. Ubuntu内部先安装cuda11.8和cudnn
  2. 然后安装pytorch
  3. conda install packaging
  4. 然后安装pip install causal-conv1d==1.2.0,然后就报错了。也是都没等到安装Manba
    :::
    任然报错

经历三

完全按照作者提到的
:::info

  1. conda create -n your_env_name python=3.10.13
  2. conda activate your_env_name
  3. conda install cudatoolkit==11.8 -c nvidia
  4. pip install torch2.1.1 torchvision0.16.1 torchaudio==2.1.1 --index-url https://download.pytorch.org/whl/cu118
  5. conda install -c “nvidia/label/cuda-11.8.0” cuda-nvcc
  6. conda install packaging
  7. pip install causal-conv1d==1.2.0 # 此处报错
  8. pip install mamba-ssm
    :::
    image.png

前面已经安装好了很多的依赖,只不过还是报错了
说是链接超时,网络问题。看到了希望

经历4

conda create -n your_env_name python=3.10.13
conda activate your_env_name
conda install cudatoolkit==11.8 -c nvidia
pip install torch==2.1.1 torchvision==0.16.1 torchaudio==2.1.1 --index-url https://download.pytorch.org/whl/cu118
conda install -c "nvidia/label/cuda-11.8.0" cuda-nvcc
conda install packaging
git clone https://github.com/Dao-AILab/causal-conv1d.git 
cd causal-conv1d 
git checkout v1.2.0 # current latest version tag 
CAUSAL_CONV1D_FORCE_BUILD=TRUE pip install .
pip install mamba-ssm # 此处报错

image.png
至此才有了文章顶部的解决方案

conda环境内部安装cuda

好处就是当前环境使用的cuda和机器内的cuda不冲突

conda install cudatoolkit==11.8 -c nvidia
conda install -c "nvidia/label/cuda-11.8.0" cuda-nvcc

参考:flash-attention踩坑:使用conda管理CUDA

本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处:http://www.hqwc.cn/news/511936.html

如若内容造成侵权/违法违规/事实不符,请联系编程知识网进行投诉反馈email:809451989@qq.com,一经查实,立即删除!

相关文章

基于java的母婴商城系统设计与实现

摘 要 现代经济快节奏发展以及不断完善升级的信息化技术,让传统数据信息的管理升级为软件存储,归纳,集中处理数据信息的管理方式。本母婴商城系统就是在这样的大环境下诞生,其可以帮助管理者在短时间内处理完毕庞大的数据信息&a…

如何做代币分析:以 CRO 币为例

作者:lesleyfootprint.network 编译:Mingfootprint.network 数据源:CRO Token Dashboard (仅包括以太坊数据) 在加密货币和数字资产领域,代币分析起着至关重要的作用。代币分析指的是深入研究与代币相关…

一个完整的Flutter项目的基本构成

目录 1.页面跳转2.本地数据库和读取2.1 在pubspec.yaml中添加数据库框架依赖2.2 创建db.dart 初始化数据库并创建表2.3 安装JsonToDart插件2.4 创建实体类 user_bean.dart2.5 增删改查: 3.网络请求数据解析UI渲染 本篇主要总结下一个完整的Flutter项目有哪些基本构成…

常用“树”数据结构

哈夫曼树 在许多应用中,树中结点常常被赋予一个表示某种意义的数值,称为该结点的权。从树的根到任意结点的路径长度(经过的边数)与该结点上权值的乘积,称为该结点的带权路径长度。树中所有叶结点的带权路径长度之和称为该树的带权路径长度&am…

hot100 -- 普通数组

目录 🎂最大子数组和 O(n) 暴力 O(n) 动态规划 🚩合并区间 O(nlogn) 排序 🌼轮转数组 O(n) 辅助数组 O(n) 环状替换 O(n) 数组翻转 🌼除自身以外数组的乘积 O(n) 前缀和 时间O(n) 空间O(1) 🌙缺失的…

波奇学Linux:信号的产生:异常和软件条件

异常与信号 进程异常的本质是程序收到操作系统信号 eg:除0让进程收到异常信号,kill掉进程,使得进程崩溃 进程收到异常信号不一定会退出,对出程序异常,捕捉信号,进程可能不退出。 操作系统如何知道除0操作…

javaWebssh题库管理系统myeclipse开发mysql数据库MVC模式java编程计算机网页设计

一、源码特点 java ssh题库管理系统是一套完善的web设计系统(系统采用ssh框架进行设计开发),对理解JSP java编程开发语言有帮助,系统具有完整的源代码和数据库,系统主要采用B/S模式开发。开发环境为TOMCAT7.0,Mye…

折线图实现柱状阴影背景的demo

这个是一个由官网的基础折线图实现的流程,将涉及到的知识点附上个人浅薄的见解,源码在最后,需要的可自取。 折线图 成果展示代码注解参数backgroundColordataZoomlegendtitlexAxisyAxisgridseries 源码 成果展示 官网的基础折线图&#xff…

Jupyter Notebook的安装和使用(windows环境)

一、jupyter notebook 安装 前提条件:安装python环境 安装python环境步骤: 1.下载官方python解释器 2.安装python 3.命令行窗口敲击命令pip install jupyter 4.安装jupyter之后,直接启动命令jupyter notebook,在默认浏览器中打开jupyte…

【译】WordPress Bricks主题安全漏洞曝光,25,000个安装受影响

WordPress的Bricks主题存在一个严重的安全漏洞,恶意威胁行为者正在积极利用该漏洞在易受攻击的安装上运行任意PHP代码。 该漏洞被跟踪为CVE-2024-25600(CVSS评分:9.8),使未经身份验证的攻击者能够实现远程代码执行。它影响Bricks的所有版本,包括1.9.6版本及更早版本。 该…

今日Arxiv最热NLP大模型论文:北京大学发布“讨论链”,利用LLM协作回答复杂问题

开放式问答(Open-ended Question Answering, QA)是人工智能领域的一个重要分支,它要求模型能够找到合适的证据,并形成有理有据、全面且有帮助的答案。在实际应用中,模型还需要能够就与问题紧密相关的潜在场景进行扩展讨…

python打包两种方式:setup.py、pyproject.toml;entry_points、project.scripts 可执行的命令行

参考: https://blog.csdn.net/qq_38844437/article/details/126628564 setup.py https://click.palletsprojects.com/en/7.x/setuptools/#setuptools-integration 学习写法:https://github.com/myshell-ai/MeloTTS/blob/main/setup.py yourscript.p…