Win10 下 Vision Mamba（Vim-main）的环境配置（libcuda.so文件无法找到，windows系统运行失败）_vim mamba

目录

1、下载NVIDIA 驱动程序、cuda11.8、cudnn8.6.0

2、在Anaconda中创建环境并激活

3、下载gpu版本的torch

4、配置环境所需要的包

5、安装causal_conv1d和mamba-1p1p1

安装causal_conv1d

安装mamba-1p1p1

6、运行main.py失败  

---

请直接拉到最后查看运行失败的原因，如有解决方案欢迎讨论

1、下载NVIDIA 驱动程序、cuda11.8、cudnn8.6.0

参考另一篇博客【win10+cuda11.8+cudnn8.6.0安装】

2、在Anaconda中创建环境并激活

• 创建环境

环境名：vim，python版本：3.10.13

conda create -n vim python=3.10.13

• 激活环境

conda activate vim

3、下载gpu版本的torch

pip install torch==2.1.1 torchvision==0.16.1 torchaudio==2.1.1 --index-url https://download.pytorch.org/whl/cu118

4、配置环境所需要的包

如果安装过程中，有部分包指定的版本安装失败，则直接【pip install package】，不指定版本安装即可。

pip install -r vim/vim_requirements.txt

• 官方提供的triton包只支持Linux系统，请查看windows安装triton的教程链接。

5、安装causal_conv1d和mamba-1p1p1

这两个包作者提供的源码里面有，目录结构如下：

• 安装causal_conv1d

进入VIm-main项目文件夹的causal-conv1d文件夹后，利用-e参数安装。

cd causal-conv1d
pip install -e .

安装caucal_conv1d的时候，报错：

error: Microsoft Visual C++ 14.0 or greater is required. Get it with “Microsoft C++ Build Tools”: https://visualstudio.microsoft.com/visual-cpp-build-tools/

解决方案参考：【已解决】error: Microsoft Visual C++ 14.0 or greater is required

后来发现安装完的版本是1.0.0，但是要求>=1.1.0，所以重新自己下载1.1.3版本之后再安装

causal_conv1d的1.1.3版本的下载链接：GitCode - 开发者的代码家园

下载完之后，进入causal_conv1d文件夹，输入命令【pip install -e .】安装即可。

安装完后，使用命令【pip list】查看如下：

• 安装mamba-1p1p1

windows无法使用【MAMBA_FORCE_BUILD=TRUE pip install .】命令安装 mamba-ssm， 需要把Vim-main/mamba-1p1p1/setup.py文件中的两个变量改成 False（原因：改成False后才能强制利用已经下载的本地文件进行编译）。修改的内容如下：

修改完后，在cmd进入mamba-1p1p1文件下，执行安装。

cd mamba-1p1p1
pip install -e .

安装成功的话，显示如下

报错：【ModuleNotFoundError: No module named 'selective_scan_cuda'】

解决办法：

找到文件 mamba-1p1p1/mamba_ssm/ops/selective_scan_interface.py，修改三处内容

第一处，注释import语句

# import selective_scan_cuda

第二处，将原来的return语句注释掉，更换成新的return语句

# 将原来的return语句注释掉，更换成新的return语句
def selective_scan_fn(u, delta, A, B, C, D=None, z=None, delta_bias=None, delta_softplus=False,
                     return_last_state=False):
    """if return_last_state is True, returns (out, last_state)
    last_state has shape (batch, dim, dstate). Note that the gradient of the last state is
    not considered in the backward pass.
    """
    # return SelectiveScanFn.apply(u, delta, A, B, C, D, z, delta_bias, delta_softplus, return_last_state)
    return selective_scan_ref(u, delta, A, B, C, D, z, delta_bias, delta_softplus, return_last_state)  # here

 第三处，将原来的return语句注释掉，更换成新的return语句

# 将原来的return语句注释掉，更换成新的return语句
def mamba_inner_fn(
    xz, conv1d_weight, conv1d_bias, x_proj_weight, delta_proj_weight,
    out_proj_weight, out_proj_bias,
    A, B=None, C=None, D=None, delta_bias=None, B_proj_bias=None,
    C_proj_bias=None, delta_softplus=True
):
    # return MambaInnerFn.apply(xz, conv1d_weight, conv1d_bias, x_proj_weight, delta_proj_weight,
    #                           out_proj_weight, out_proj_bias,
    #                           A, B, C, D, delta_bias, B_proj_bias, C_proj_bias, delta_softplus)
    return mamba_inner_ref(xz, conv1d_weight, conv1d_bias, x_proj_weight, delta_proj_weight,
                           out_proj_weight, out_proj_bias,
                           A, B, C, D, delta_bias, B_proj_bias, C_proj_bias, delta_softplus)  # here

6、运行main.py失败  

报错1：

File "D:\anaconda\install\envs\vim\lib\site-packages\triton\compiler.py", line 1093, in default_cache_dir return os.path.join(os.environ["HOME"], ".triton", "cache") File "D:\anaconda\install\envs\vim\lib\os.py", line 680, in __getitem__ raise KeyError(key) from None KeyError: 'HOME' 

尝试解决：

在D盘下新建了一个HOME目录，然后将HOME环境变量设置为D:\HOME。关闭PyCharm并重启，使环境变量生效。重新运行main.py之后，没有报这个错误，报了另一个error。

---

报错2：
  File "D:\anaconda\install\envs\vim\lib\site-packages\triton\runtime\autotuner.py", line 62, in 
  File "D:\anaconda\install\envs\vim\lib\site-packages\triton\compiler.py", line 1155, in _build
    cuda_lib_dirs = libcuda_dirs()
  File "D:\anaconda\install\envs\vim\lib\site-packages\triton\compiler.py", line 1134, in libcuda_dirs
    locs = subprocess.check_output(["whereis", "libcuda.so"]).decode().strip().split()[1:]
  File "D:\anaconda\install\envs\vim\lib\subprocess.py", line 421, in check_output
    return run(*popenargs, stdout=PIPE, timeout=timeout, check=True,
  File "D:\anaconda\install\envs\vim\lib\subprocess.py", line 503, in run
    with Popen(*popenargs, **kwargs) as process:
  File "D:\anaconda\install\envs\vim\lib\subprocess.py", line 971, in __init__
    self._execute_child(args, executable, preexec_fn, close_fds,
  File "D:\anaconda\install\envs\vim\lib\subprocess.py", line 1456, in _execute_child
    hp, ht, pid, tid = _winapi.CreateProcess(executable, args,
FileNotFoundError: [WinError 2] 系统找不到指定的文件。

原因分析： triton 库出现了问题，似乎是在尝试执行 whereis libcuda.so 命令时发生了问题，因为在 Windows 系统中并不存在 libcuda.so 文件。待解决...