适用:Linux 服务器,4卡GPU环境,实现模型多卡分片推理,CUDA多卡分发
一键脚本安装(也可使用压缩包手动安装)
bash展开代码curl -fsSL [https://ollama.com/install.sh](https://ollama.com/install.sh) | sh
安装完成后 systemd 会自动后台启动 ollama 服务。
bash展开代码ollama -v
执行编辑命令:
bash展开代码sudo systemctl edit ollama
在弹出编辑器中粘贴下面配置:
ini展开代码[Service]
Environment="CUDA_VISIBLE_DEVICES=0,1,2,3"
Environment="OLLAMA_SCHED_SPREAD=1"
Ctrl+O 保存Ctrl+X 退出说明:该命令生成override覆盖配置,不会修改原始 ollama.service 文件
bash展开代码sudo systemctl daemon-reload sudo systemctl restart ollama
⚠️ 重要提醒:已经加载在单卡显存里的模型不会自动重新分片!需要卸载旧模型,重新拉取加载,多卡分片才会生效。
⭐ 必须先卸载旧模型,再重新拉取,多卡分片配置才能生效
bash展开代码# 删除 deepseek-r1:70b 模型
ollama rm deepseek-r1:70b
补充:
ollama rm:remove,删除本地模型文件- 执行后模型文件会从磁盘清除,需要重新 pull 下载
默认拉取 GGUF Q4_K_M 版本(约40GB)
bash展开代码ollama pull deepseek-r1:70b
可选更高精度版本 Q8_0(75GB,显存压力更大)
bash展开代码ollama pull deepseek-r1:70b-q8_0
查看本地已下载模型列表
bash展开代码ollama list
bash展开代码ollama run deepseek-r1:70b
进入交互界面后可直接提问。
展开代码/set parameter num_ctx 8192
展开代码/exit
新开终端执行
bash展开代码nvidia-smi
4张A6000显卡均占用显存,代表多卡分片成功。
OLLAMA_SCHED_SPREAD=1 开启分片调度,配合CUDA_VISIBLE_DEVICES指定可用显卡。本文作者:zzz
本文链接:
版权声明:本博客所有文章除特别声明外,均采用 BY-NC-SA 许可协议。转载请注明出处!