- 安装 NVIDIA 驱动与 CUDA
最稳妥的方式是使用 Ubuntu 的官方源来安装驱动,它能自动处理依赖关系,避免冲突。
更新系统并安装驱动工具
bash
sudo apt update
sudo apt upgrade -y
sudo apt install -y ubuntu-drivers-common build-essential
自动安装推荐驱动(最简单)
这条命令会自动检测你的显卡并安装最合适的驱动版本。
bash
sudo ubuntu-drivers autoinstall
安装完成后,务必重启系统:
bash
sudo reboot
验证驱动安装
重启后,运行 nvidia-smi,你应该能看到 GPU 信息、驱动版本和最高支持的 CUDA 版本。例如,输出可能会显示 CUDA Version: 12.4。
安装 CUDA 工具包
为了让 llama.cpp 在编译时能用到 CUDA,需要安装开发工具包。通过 NVIDIA 官方网络源安装是最简单的方式。
bash
# 下载并安装 CUDA 密钥环
wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2404/x86_64/cuda-keyring_1.1-1_all.deb
sudo dpkg -i cuda-keyring_1.1-1_all.deb
# 更新源并安装 CUDA 工具包
sudo apt update
sudo apt install -y cuda-toolkit
配置环境变量
将 CUDA 添加到 PATH,方便后续使用。
bash
echo 'export PATH=/usr/local/cuda/bin${PATH:+:${PATH}}' >> ~/.bashrc
echo 'export LD_LIBRARY_PATH=/usr/local/cuda/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}' >> ~/.bashrc
source ~/.bashrc
最后,运行 nvcc --version 验证 CUDA 编译器是否安装成功。
- 安装与编译 llama.cpp
llama.cpp 支持从源码编译,这样能确保你的机器获得最优性能。
安装编译依赖
bash
sudo apt install -y cmake git
克隆源码
bash
git clone https://github.com/ggerganov/llama.cpp
cd llama.cpp
创建构建目录并编译
使用 CMake 进行编译。关键在于通过 -DGGML_CUDA=ON 开启 CUDA 支持。-j 参数可以加快编译速度。
bash
mkdir build && cd build
cmake .. -DCMAKE_BUILD_TYPE=Release -DGGML_CUDA=ON
cmake --build . --config Release -j $(nproc)
编译完成后,可执行文件(如 llama-server, llama-cli)就在 build/bin/ 目录下