Mamba windows 环境安装踩坑 ModuleNotFoundError: No module named ‘causal_conv1d_cuda‘|‘selective_scan_cuda‘

华魔王与青衫客

已于 2024-06-02 01:41:09 修改

阅读量9k

点赞数 29

文章标签： windows python 计算机视觉语言模型超分辨率重建 microsoft

于 2024-06-02 01:09:30 首次发布

本文链接：https://blog.csdn.net/weixin_51005487/article/details/139382288

版权

电脑环境：

windows11 cuda11.6(最高只到11.6了，呜呜)

torch 1.13 python 3.10

这里有一个点需要注意，安装mamba的时候需要安装一个依赖包 triton ,本来triton 没有windows版，于是github上有大佬编译了triton-windows版本，我去查看的时候只有适配py3.10版本的（这个点我记得比较清楚，当时我搜索的时候只有适配 py3.1 0的，为此我安装了python3.10）。

目前triton更新了多少，我并不清楚，所以最好提前搜索一下 trition 的wheel下载，确定一下现在有多少适配的版本，如果适配现有python版本的话就不用换python版本了，省了一个步骤。

回到正题，对于mamba安装，在github上只能找到cu118的wheel, 所以我选择的是源代码安装，方法参考下面的文章（在此说明，我并不清楚cu116的环境，下载cu118的wheel是不是能安装使用，不过你们可以试试，最好反馈一下，如果可以的话，下次我就不用这么麻烦了）

安装以后就会