+-----------------------------------------------------------------------------------------+ | Processes: | | GPU GI CI PID Type Process name GPU Memory | | ID ID Usage | |=========================================================================================| | No running processes found | +-----------------------------------------------------------------------------------------+
注意:nvidia-smi 命令显示 CUDA UMD Version: 13.3 的意思是显卡驱动支持的 CUDA Runtime API 版本最高为 13.3,不代表 WSL 2 中存在 CUDA Runtime。
CUDA Toolkit
CUDA Runtime 以及编译 llama.cpp 需要用到的 CUDA 编译器 nvcc,都需要通过安装 CUDA Toolkit 来得到:
# 1. 拉取代码并进入项目目录 $ git clone https://github.com/ggml-org/llama.cpp $ cd llama.cpp/
# 2.a 根据本机环境生成构建文件,启用 CUDA 后端支持 # -DGGML_CUDA=ON:启用 GGML 张量计算库的 CUDA 后端支持 $ cmake -B build -DGGML_CUDA=ON
# 3. 执行编译 $ cmake --build build --config Release
# 编译得到的可执行文件存放在 build/bin 下面 $ cd build/bin/ # 测试 llama-cli 命令 $ ./llama-cli --version version: 9859 (4fc4ec554) built with GNU 13.3.0 for Linux x86_64
# 在不同 GPU 服务器执行 $ nvidia-smi --query-gpu=name,compute_cap --format=csv name, compute_cap Tesla V100-PCIE-16GB, 7.0 Tesla V100-PCIE-16GB, 7.0 Tesla V100-PCIE-16GB, 7.0 Tesla V100-PCIE-16GB, 7.0 $ nvidia-smi --query-gpu=name,compute_cap --format=csv name, compute_cap Tesla T4, 7.5 Tesla T4, 7.5 Tesla T4, 7.5 Tesla T4, 7.5