mamba_ssm-2.2.2-cp310-cp310-win_amd64.whl

作品简介

mambassm-2.2.2-cp310-cp310-winamd64.whl+安装环境+测试脚本.7z提供百度云盘地址下载

编译环境:

vs2022

win10 x64

anaconda3+python3.10

torch==2.3.1+cu118

cuda11.8.0+cudnn8.9.7

triton==2.1.0

causal_conv1d==1.4.0

mamba==2.2.2

RTX2070显卡

注意编译的whl是不能用于RTX50显卡的,可以用于RTX20-RTX40系列显卡,安装时候尽量和模块一致

创作时间: