当前位置: 首页 > news >正文

Ubuntu24.04环境下causal_conv1d和mamba_ssm安装

环境:WSL的Ubuntu24.04

1.创建conda环境,其中python版本为3.10.13

2.当前conda环境依次执行下面命令:

conda install cudatoolkit==11.8 -c nvidia
pip install torch==2.1.1 torchvision==0.16.1 torchaudio==2.1.1 -f https://mirrors.aliyun.com/pytorch-wheels/cu118
conda install -c "nvidia/label/cuda-11.8.0" cuda-nvcc
conda install packaging

3.安装causal_conv1d和mamba_ssm

3.1尝试使用pip线上安装causal_conv1d:

pip install causal-conv1d==1.5.2

失败。
3.2尝试本地安装:

https://github.com/Dao-AILab/causal-conv1d/releases/tag/v1.5.0

下载

causal_conv1d-1.5.0+cu118torch2.1cxx11abiFALSE-cp310-cp310-linux_x86_64.whl

https://github.com/state-spaces/mamba/releases/tag/v2.0.3

下载

mamba_ssm-2.0.3+cu118torch2.1cxx11abiFALSE-cp310-cp310-linux_x86_64.whl

注意这两个whl文件名中的“cu”后的数字(即cuda版本)要与上一步安装的cuda版本相同,“torch”后面的数字要与上一步安装的torch版本相同,“cp”后面的数字要与你当前环境的python版本相同。
在这两个whl文件所在目录下分别执行:

pip install 文件名

根据打印信息确定是否安装成功。
python导入:

from causal_conv1d import causal_conv1d_fn, causal_conv1d_update

但打印如下信息:
from causal_conv1d import causal_conv1d_fn, causal_conv1d_update

A module that was compiled using NumPy 1.x cannot be run in
NumPy 2.2.6 as it may crash. To support both 1.x and 2.x
versions of NumPy, modules must be compiled with NumPy 2.0.
Some module may need to rebuild instead e.g. with 'pybind11>=2.12'.

If you are a user of the module, the easiest solution will be to
downgrade to 'numpy<2' or try to upgrade the affected module.
We expect that some modules will need time to support NumPy 2.
...
/miniconda3/envs/mapy31013/lib/python3.10/site-packages/torch/nn/modules/transformer.py:20: UserWarning: Failed to initialize NumPy: _ARRAY_API not found (Triggered internally at ../torch/csrc/utils/tensor_numpy.cpp:84.)
device: torch.device = torch.device(torch._C._get_default_device()),  # torch.device('cpu'),
就是说当前虚拟环境的NumPy版本为numpy2+,需要使用numpy<2版本才能导入causal_conv1d。
于是先

pip uninstall numpy

https://pypi.org/project/numpy/

网页的左侧点击“Release history”找到合适的numpy版本(我选的是1.26.4),再执行

pip install numpy==1.26.4

顺利安装numpy1.26.4后再次执行

from causal_conv1d import causal_conv1d_fn, causal_conv1d_update

不报错则安装成功。

3.3python导入:

from mamba_ssm import Mamba, Mamba2

不报错则mamba安装成功。

http://www.dtcms.com/a/348073.html

相关文章:

  • 嵌入式八股文面试题总结(QT、RTOS、Linux、ARM、C/C++)(持续更新)
  • QT-布局管理器
  • 音视频面试题集锦第 32 期
  • C语言指针5
  • 使用虚幻引擎5(UE5)开发类似《原神》的开放世界游戏:从技术架构到实践指南
  • LeetCode-542. 01 矩阵
  • (LeetCode 每日一题) 1493. 删掉一个元素以后全为 1 的最长子数组 (双指针)
  • 03-鸿蒙架构与编程模型
  • ChainVault:重塑亚洲黄金交易基建,引领RWA金融新浪潮
  • Java 22 新特性及具体应用
  • week4-[字符数组]英语作文
  • Games101 作业1 旋转与投影
  • 量子链(Qtum)分布式治理协议
  • Qt+windows+QtInstallerFramework打包可执行文件教程-美观且简单
  • Django的Serializers与 fastapi 的Pydantic
  • Excel 转化成JSON
  • Java:LinkedList的使用
  • Django的Settings 配置文件详解
  • 【ArcGIS Pro 全攻略】GIS 数据格式终极指南:从原理到实战,再也不纠结选哪种格式!
  • React useState 全面深入解析
  • Linux 824 shell:expect
  • 基于5G NR NTN与DVB-S2X/RCS2的机载卫星通信终端性能分析
  • 低功耗模式DMA数据搬运问题解析
  • 在测试接入抖音小游戏订阅消息推送时遇到的问题
  • bun + vite7 的结合,孕育的 Robot Admin 【靓仔出道】(十八)
  • K8s部署MySQL8.0数据库
  • Transformer实战(13)——从零开始训练GPT-2语言模型
  • 【go语言】字符串函数
  • imx6ull-驱动开发篇39——Linux INPUT 子系统实验
  • 05-ArkUI界面开发