- 博客(1)
- 收藏
- 关注
原创 从零开始在Jetson硬件上配置pytorch和mamba环境
本文详细记录了在英伟达Jetson Xavier NX设备上配置Mamba环境的完整过程。硬件配置为Jetpack-5.1.6、Ubuntu-20.04、CUDA-11.8,软件环境包括Python-3.8.6、PyTorch-2.1.0等。文章从刷机开始,详细介绍了PyTorch安装、CUDA升级到11.8、LLVM和Triton的编译安装,以及causal-conv1d和mamba-ssm的安装过程。重点解决了ARM架构(aarch64)下的兼容性问题,包括修改setup.py文件、处理内存溢出、设置环
2026-03-20 22:52:22
698
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人
RSS订阅