在Ubuntu系统下使用Docker部署ollama并使用英伟达gpu加速
本文介绍了英伟达Nvidia容器工具包的安装步骤:首先通过curl命令下载并添加GPG密钥和软件源列表,然后更新apt软件包索引并安装指定版本的nvidia-container-toolkit及相关组件。安装完成后需要重启Docker服务(sudo systemctl restart docker)。最后展示了如何运行支持GPU的Docker容器,示例中使用ollama镜像,配置了GPU支持、数据
·
下载安装英伟达Nvidia container toolkit:
curl -fsSL https://nvidia.github.io/libnvidia-container/gpgkey | sudo gpg --dearmor -o /usr/share/keyrings/nvidia-container-toolkit-keyring.gpg \
&& curl -s -L https://nvidia.github.io/libnvidia-container/stable/deb/nvidia-container-toolkit.list | \
sed 's#deb https://#deb [signed-by=/usr/share/keyrings/nvidia-container-toolkit-keyring.gpg] https://#g' | \
sudo tee /etc/apt/sources.list.d/nvidia-container-toolkit.list
sudo sed -i -e '/experimental/ s/^#//g' /etc/apt/sources.list.d/nvidia-container-toolkit.list
sudo apt-get update
export NVIDIA_CONTAINER_TOOLKIT_VERSION=1.17.8-1
sudo apt-get install -y \
nvidia-container-toolkit=${NVIDIA_CONTAINER_TOOLKIT_VERSION} \
nvidia-container-toolkit-base=${NVIDIA_CONTAINER_TOOLKIT_VERSION} \
libnvidia-container-tools=${NVIDIA_CONTAINER_TOOLKIT_VERSION} \
libnvidia-container1=${NVIDIA_CONTAINER_TOOLKIT_VERSION}
安装完成后检查是否安装成功
nvidia-container-runtime --version
安装完Nvidia container toolkit 后重启docker :
sudo systemctl restart docker
运行docker容器:
sudo docker run -d --gpus=all -v ollama:/root/.ollama -p 11434:11434 --name ollama ollama/ollama
更多推荐
所有评论(0)