MonoGS项目运行中的CUBLAS兼容性问题解决方案

问题背景

在使用MonoGS项目进行3D场景重建时，部分用户在运行slam.py脚本时遇到了CUBLAS_STATUS_NOT_SUPPORTED错误。这个错误通常发生在CUDA矩阵乘法运算(cublasSgemmStridedBatched)调用过程中，表明当前CUDA环境与项目所需的计算库存在兼容性问题。

错误分析

该错误的核心是CUDA计算库版本不匹配导致的兼容性问题。具体表现为：

1. 错误发生在矩阵乘法运算的核心计算阶段
2. 涉及CUDA的BLAS库(cublas)功能调用
3. 与操作系统版本和CUDA工具包版本密切相关

解决方案

根据项目维护者和社区的经验，这个问题可以通过调整CUDA工具包版本来解决：

针对Ubuntu 18.04系统

推荐使用CUDA 11.3工具包版本，可以通过以下步骤配置：

1. 创建新的conda环境
2. 安装指定版本的CUDA工具包
3. 安装匹配版本的PyTorch

针对Ubuntu 20.04系统

需要升级到CUDA 11.6工具包版本，配置步骤类似但版本号不同。

详细配置步骤

以下是经过验证的完整环境配置流程：

1. 创建conda环境

conda create -n monogs python=3.7.13
conda activate monogs

2. 安装CUDA工具包

conda install -c "nvidia/label/cuda-11.6.0" cuda-toolkit

3. 安装PyTorch

conda install pytorch==1.12.1 torchvision==0.13.1 torchaudio==0.12.1 cudatoolkit=11.6 -c pytorch -c conda-forge

4. 安装项目依赖

pip install -r requirements.txt

5. 编译子模块

cd submodules/diff-gaussian-rasterization/
python setup.py install
pip install .
cd simple-knn/
python setup.py install
pip install .

注意事项

1. CUDA工具包安装可能需要较长时间(约1小时)
2. 确保所有组件的版本匹配，特别是PyTorch与CUDA的版本对应关系
3. 不同操作系统可能需要不同的CUDA版本组合

总结

MonoGS项目对CUDA环境有特定要求，正确配置CUDA工具包版本是解决CUBLAS_STATUS_NOT_SUPPORTED错误的关键。用户应根据自己的操作系统版本选择合适的CUDA版本组合，并确保所有相关组件版本兼容。通过上述步骤配置环境后，项目应能正常运行。