在linux上安装ollama

您所在的位置：首页显卡知识

在linux上安装ollama
阅读量：100 发布时间：2024-12-03 08:12:41

一. Centos7安装NVIDIA显卡驱动

[root@jxgm7 docker]# lspci | grep -i vga

00:02.0 VGA compatible controller: Intel Corporation UHD Graphics 630 (Mobile)

01:00.0 VGA compatible controller: NVIDIA Corporation GP107M [GeForce GTX 1050 Ti Mobile] (rev a1)

有个核显，一个独显

二.更新系统并安装依赖

yum update kernel

yum install kernel-devel gcc -y

reboot

三.屏蔽掉nouveau

这个结合了网上的两点,

修改dist-blacklist.conf文件

vim /lib/modprobe.d/dist-blacklist.conf

将nvidiafb注释掉:

#blacklist nvidiafb

然后添加以下语句：

blacklist nouveau

options nouveau modeset=0

修改/etc/default/grub 文件

重建initramfs image

备份 mv /boot/initramfs-$(uname -r).img /boot/initramfs-$(uname -r).img.bak

重建 dracut /boot/initramfs-$(uname -r).img $(uname -r)

重启 reboot

四.安装NVIDIA驱动

1.在NVIDIA官网下载驱动

网址
:https://www.nvidia.cn/Download/index.aspx?lang=cn

我下载的是
NVIDIA-Linux-x86_64-550.54.14.run

2.切换到命令行界面

init 3

3.安装NVIDIA驱动

chmod +x NVIDIA-Linux-x86_64-550.54.14.run

./NVIDIA-Linux-x86_64-550.54.14.run

查看安装结果

nvidia-smi

五.用docker安装ollama

1.centos安装docker-ce

# 设置docker镜像源
yum-config-manager \
    --add-repo \
    https://mirrors.aliyun.com/docker-ce/linux/centos/docker-ce.repo
    
sed -i 's/download.docker.com/mirrors.aliyun.com\/docker-ce/g' /etc/yum.repos.d/docker-ce.repo

yum makecache fast

yum install -y docker-ce

2.安装NVIDIA Container Toolkit.

yum install -y nvidia-container-toolkit
nvidia-ctk runtime configure --runtime=docker
systemctl restart docker

3.安装ollama镜像

docker run -d --gpus=all -v /root/docker/ollama:/root/.ollama -p 11434:11434 --name ollama ollama/ollama

4.安装openweb-ui

docker run -d -p 3000:8080 --add-host=host.docker.internal:host-gateway -v /root/docker/open-webui:/app/backend/data --name open-webui --restart always ghcr.io/open-webui/open-webui:main

5.下载模型等

我这里下载了llama3.1与qwen2等模型,

执行命令是

docker exec -it ollama ollama run llama3.1

docker exec -it ollama ollama run qwen2

打开http://ip:3000之后,登录完选择模型,即可对话了