查看cuda版本:
nvcc --version
本文涉及到的需要安装的git、conda等软件安装与配置请移步至:创作中心-CSDN
1、克隆仓库
git clone https://github.com/manycore-research/SpatialLM.git
cd SpatialLM
运行结果:
2、创建虚拟环境
conda create -n spatiallm python=3.11
conda activate spatiallm
conda install -y nvidia/label/cuda-12.4.0::cuda-toolkit conda-forge::sparsehash
使用conda创建虚拟环境:
激活环境并安装cuda依赖:
3、安装poetry包管理工具
-
poetry config virtualenvs.create false --local:
禁用 Poetry 自动创建虚拟环境,将配置应用于当前项目(而不是全局配置)。 说明:默认情况下,Poetry 会为每个项目创建一个虚拟环境。如果你希望直接使用全局 Python 环境(例如,你已经配置了虚拟环境管理工具,如venv
或conda
),可以禁用 Poetry 的自动虚拟环境创建功能。
pycharm的终端为cmd.exe时,运行:
pip install poetry && poetry config virtualenvs.create false --local
pycharm的终端为PowerShell时,运行:
pip install poetry; poetry config virtualenvs.create false --local
4、安装依赖
poetry install
-
作用:
读取项目的pyproject.toml
文件,解析项目所需的依赖。安装所有列出的依赖,并将它们的版本锁定到poetry.lock
文件中。根据配置决定是否创建虚拟环境。 -
如果项目配置了虚拟环境(默认行为),它会创建一个虚拟环境,并在其中安装依赖。
-
如果禁用了虚拟环境(如
poetry config virtualenvs.create false
),它会直接在全局 Python 环境中安装依赖。
Installing nvidia-nccl-cu12 (2.20.5): Failed:
nccl
验证安装,列出项目的所有依赖及其版本:
poetry show
运行以下代码将会报错:
poe install-torchsparse
解决方案:手动下载并安装
访问 https://github.com/mit-han-lab/torchsparse → 点击 Code
→ 选择 Download ZIP
解压 ZIP 文件到本地目录(如 D:\torchsparse
)
通过 pip 从本地安装:
pip install D:\torchsparse
上述安装torchsparse时,依赖库
pygments安装失败,替代方案:
pip install pygments --index-url https://pypi.tuna.tsinghua.edu.cn/simple
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/