建议使用 Linux 系统或 Mac 系统,Windows 下使用 WSL(Windows Subsystem for Linux),配置方法与 Linux 一致。
sudo apt update
sudo apt install gcc g++brew install gccsudo apt install cmakebrew install cmake需通过 CUDA Toolkit 安装
# 官网下载对应系统的 CUDA Toolkit 安装包
# 示例(Ubuntu 22.04):
wget https://developer.download.nvidia.com/compute/cuda/12.6.1/local_installers/cuda_12.6.1_550.54.15_linux.run
sudo sh cuda_12.6.1_550.54.15_linux.run- 访问 CUDA Toolkit 官网下载安装包
- 安装后检查版本:
nvcc --version
sudo apt install makebrew install makeNVIDIA官方提供的docker镜像支持编译本项目的环境,推荐拉取nvcr.io/nvidia/pytorch:25.06-py3。docker的具体操作方法不在此展开。
gcc --version # 检查 GCC ≥13
g++ --version # 检查 G++ ≥13
cmake --version # 检查 CMake ≥3.13
nvcc --version # 检查 NVCC ≥12.6(建议,低版本可能支持)
make --version配置完成后进入项目目录,执行以下命令:
make build # 构建整个项目,默认支持cuda编译
make build USE_CUDA=OFF # 显示关闭cuda编译
make test-cpp # 构建并运行测试
make clean # 清理生成文件