1. 安装triton,下面提供一个WIN10/11编译好的Triton版本:
triton-2.0.0-cp310-cp310-win_amd64.whl
运行指令:pip install triton-2.0.0-cp310-cp310-win_amd64.whl
2. 安装causal-conv1d
FORCE_BUILD = os.getenv("CAUSAL_CONV1D_FORCE_BUILD", "FALSE") == "TRUE"
SKIP_CUDA_BUILD = os.getenv("CAUSAL_CONV1D_SKIP_CUDA_BUILD", "FALSE") == "TRUE"
# For CI, we want the option to build with C++11 ABI since the nvcr images use C++11 ABI
FORCE_CXX11_ABI = os.getenv("CAUSAL_CONV1D_FORCE_CXX11_ABI", "FALSE") == "TRUE"
将三个TRUE改为FALSE:
FORCE_BUILD = os.getenv("CAUSAL_CONV1D_FORCE_BUILD", "FALSE") == "FALSE"
SKIP_CUDA_BUILD = os.getenv("CAUSAL_CONV1D_SKIP_CUDA_BUILD", "FALSE") == "FALSE"
# For CI, we want the option to build with C++11 ABI since the nvcr images use C++11 ABI
FORCE_CXX11_ABI = os.getenv("CAUSAL_CONV1D_FORCE_CXX11_ABI", "FALSE") == "FALSE"
3. 问题3:RuntimeError: mamba_ssm is only supported on CUDA 11.6 and above.
直接升级升级到11.8: conda install cudatoolkit==11.8 -c nvidia
安装NVCC:conda install -c "nvidia/label/cuda-11.8.0" cuda-nvcc
或者官方下载手动安装:CUDA Toolkit 11.8 Downloads
4. 问题4:The detected CUDA version (11.8)mismatches the version that was used compile PyTorch (12.4) :
安装对应版本torch,重新卸载安装一下torch解决。
测试:
import torch
torch.cuda.is_available()
torch.backends.cudnn.version()
torch.version.cuda
5. 问题5: subprocess.CalledProcessError: Command ‘[‘ninja‘, ‘-v‘]‘ returned non-zero exit status 1.
def _run_ninja_build(build_directory: str, verbose: bool, error_prefix: str) -> None:command = ['ninja', '-v']
改成:
def _run_ninja_build(build_directory: str, verbose: bool, error_prefix: str) -> None:command = ['ninja', '--version']
6. 问题6:执行安装mamba的时候提示:error: command 'D:\\Program Files\\Microsoft Visual Studio\\2022\\Professional\\VC\\Tools\\MSVC\\14.37.32822\\bin\\HostX86\\x64\\link.exe' failed with exit code 1181
官网下载安装C++编译环境:Microsoft C++ 生成工具 - Visual Studio
参考资料:
1.Pytorch的gpu配置-CSDN博客
2.Windows系统下安装causal_conv1d和mamba_ssm_windows安装causal-conv1d包buildtool-CSDN博客
3. Python3 安装软件出现 cl.exe failed with exit status 2 错误_cl.exe' failed with exit code 2-CSDN博客
4.Windows系统下安装causal_conv1d和mamba_ssm_windows安装causal-conv1d包buildtool-CSDN博客
5. UserWarning: Error checking compiler version for cl: [WinError 2] 系统找不到指定的文件。-CSDN博客
6. Python3 安装软件出现 cl.exe failed with exit status 2 错误_cl.exe' failed with exit code 2-CSDN博客