linux安装mamba-ssm和causal-conv1d
文件，环境中装的是Python 3.8、PyTorch 2.2.2 和 CUDA 11.8。安装适用于 CUDA 11.8 的 PyTorch 2.2.2 版本。1.用conda创建一个Python 3.8 虚拟环境。3.安装 GPU 版本的 PyTorch。，选择与你的环境兼容的。
·
1.用conda创建一个Python 3.8 虚拟环境
conda create -n Mamba_py38 python=3.82.活新创建的虚拟环境：
conda activate Mamba_py383.安装 GPU 版本的 PyTorch
安装适用于 CUDA 11.8 的 PyTorch 2.2.2 版本
conda install pytorch==2.2.2 torchvision==0.17.2 torchaudio==2.2.2 pytorch-cuda=11.8 -c pytorch -c nvidia4.访问 mamba_ssm release page 和 causal-conv release page，选择与你的环境兼容的 .whl 文件，环境中装的是Python 3.8、PyTorch 2.2.2 和 CUDA 11.8。
5.使用 pip 离线安装这两个包：
pip install causal_conv1d-1.1.3+cu118torch2.2cxx11abiFALSE-cp38-cp38-linux_x86_64.whlpip install mamba_ssm-1.1.3+cu118torch2.2cxx11abiFALSE-cp38-cp38-linux_x86_64.whl
魔乐社区（Modelers.cn) 是一个中立、公益的人工智能社区，提供人工智能工具、模型、数据的托管、展示与应用协同服务，为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作，由全产业链共同建设、共同运营、共同享有，推动国产AI生态繁荣发展。
更多推荐
3
0- 0
已为社区贡献2条内容
所有评论(0)