GPUStack
GPUStack 是一個用于運行 AI 模型的開源 GPU集群管理器。它具有廣泛的硬件兼容性,支持多種品牌的 GPU,并能在 Apple MacBook、Windows PC 和 Linux 服務器上運行。GPUStack 支持各種 AI 模型,包括大型語言模型(LLMs)、擴散模型、音頻模型、嵌入模型和重新排序模型。GPUStack 可以輕松擴展,只需添加更多 GPU 或節(jié)點即可擴展操作。它支持單節(jié)點多 GPU 和多節(jié)點推理和服務,并提供多種推理后端,如llama-box、vox-box和vLLM。GPUStack 是一個輕量級的 Python 包,具有最小的依賴項和操作開銷,并且提供與 OpenAI 標準兼容的 API。此外,它還簡化了用戶和 API 密鑰的管理,提供了 GPU 性能和利用率的實時監(jiān)控,以及令牌使用和速率限制的有效跟蹤。
關(guān)鍵特性
?廣泛的硬件兼容性:支持管理 Apple Mac、Windows PC 和 Linux 服務器上不同品牌的 GPU。
?廣泛的模型支持:支持從大語言模型(LLMs)、多模態(tài)模型(VLMs)、擴散模型、語音模型到嵌入和重新排序模型的廣泛模型。
?異構(gòu) GPU 支持與擴展:能夠輕松添加異構(gòu) GPU 資源,并按需擴展算力規(guī)模。
?分布式推理:支持單機多卡并行和多機多卡并行推理。
?多推理后端支持:支持llama-box(基于 llama.cpp 和 stable-diffusion.cpp)、vox-box和vLLM作為推理后端。
?輕量級 Python 包:最小的依賴和操作開銷。
?OpenAI 兼容 API:提供兼容 OpenAI 標準的 API 服務。
?用戶和 API 密鑰管理:簡化用戶和 API 密鑰的管理流程。
?GPU 指標監(jiān)控:實時監(jiān)控 GPU 性能和利用率。
?Token 使用和速率統(tǒng)計:有效跟蹤 token 使用情況,并管理速率限制。
支持的硬件平臺
?Apple Metal(M 系列芯片)
?NVIDIA CUDA(計算能力 6.0 及以上)
?AMD ROCm
?華為昇騰(CANN)
?摩爾線程(MUSA)
?海光 DTK
支持的模型類型
?大語言模型(LLMs):如 Qwen、LLaMA、Mistral、Deepseek、Phi、Yi 等。
?多模態(tài)模型(VLMs):如 Llama3.2-Vision、Pixtral、Qwen2-VL、LLaVA、InternVL2.5 等。
?擴散模型:如 Stable Diffusion、FLUX 等。
?嵌入模型:如 BGE、BCE、Jina 等。
?重新排序模型:如 BGE、BCE、Jina 等。
?語音模型:如 Whisper(語音轉(zhuǎn)文本)、CosyVoice(文本轉(zhuǎn)語音)等。
使用場景
GPUStack 適用于需要高效管理和調(diào)度 GPU 資源的場景,特別是在運行 AI 模型時。它支持單節(jié)點多 GPU 和多節(jié)點推理及服務,并提供多種推理后端,如llama-box、vox-box和vLLM。GPUStack 是一個輕量級的 Python 包,具有最小的依賴項和操作開銷,并且提供與 OpenAI 標準兼容的 API。此外,它還簡化了用戶和 API 密鑰的管理,提供了 GPU 性能和利用率的實時監(jiān)控,以及令牌使用和速率限制的有效跟蹤。
Docker 運行 GPUStack 的詳細教程
https://docs.gpustack.ai/latest/installation/docker-installation/
以下是使用 Docker 運行 GPUStack 的詳細教程,結(jié)合官方文檔與社區(qū)實踐整理而成。通過本指南,您可以在支持 NVIDIA GPU 的 Linux 環(huán)境中快速部署 GPUStack,并實現(xiàn)異構(gòu) GPU 集群的管理與大模型服務。
在這里插入圖片描述
在這里插入圖片描述
一、環(huán)境準備
1.硬件與系統(tǒng)要求
? 確保系統(tǒng)已安裝 NVIDIA GPU,并驗證驅(qū)動兼容性(支持 CUDA 11.0 及以上版本)。
? 推薦使用 Ubuntu 22.04 LTS 或 CentOS 7+ 系統(tǒng)。
2.驗證 GPU 與依賴項
# 檢查 NVIDIA GPU 是否識別 lspci | grep -i nvidia root@i-28e6iose:/home/ubuntu# lspci | grep -i nvidia 00:0c.0 VGA compatible controller: NVIDIA Corporation TU102 [GeForce RTX 2080 Ti] (rev a1) 00:0d.0 Audio device: NVIDIA Corporation TU102 High Definition Audio Controller (rev a1) # 確認 GCC 已安裝 gcc --version root@i-28e6iose:/home/ubuntu# gcc --version gcc (Ubuntu 9.5.0-6ubuntu2) 9.5.0 Copyright (C) 2019 Free Software Foundation, Inc. This is free software; see thesourceforcopying conditions. There is NO warranty; not evenforMERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
二、安裝 NVIDIA 驅(qū)動與 Docker
1.安裝 NVIDIA 驅(qū)動
# 安裝內(nèi)核頭文件 sudoapt-get install linux-headers-$(uname-r) # 添加 CUDA 倉庫并安裝驅(qū)動 wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64/cuda-keyring_1.1-1_all.deb sudodpkg -i cuda-keyring_1.1-1_all.deb sudoapt-get update sudoapt-get install nvidia-driver-535 -y sudoreboot # 驗證驅(qū)動 nvidia-smi root@i-28e6iose:/home/ubuntu# nvidia-smi Sun Mar 9 2043 2025 +-----------------------------------------------------------------------------------------+ | NVIDIA-SMI 570.124.06 Driver Version: 570.124.06 CUDA Version: 12.8 | |-----------------------------------------+------------------------+----------------------+ | GPU Name Persistence-M | Bus-Id Disp.A | Volatile Uncorr. ECC | | Fan Temp Perf Pwr:Usage/Cap | Memory-Usage | GPU-Util Compute M. | | | | MIG M. | |=========================================+========================+======================| | 0 NVIDIA GeForce RTX 2080 Ti Off | 000000000C.0 Off | N/A | | 22% 29C P8 20W / 250W | 4MiB / 11264MiB | 0% Default | | | | N/A | +-----------------------------------------+------------------------+----------------------+ +-----------------------------------------------------------------------------------------+ | Processes: | | GPU GI CI PID Type Process name GPU Memory | | ID ID Usage | |=========================================================================================| | No running processes found | +-----------------------------------------------------------------------------------------+
2.安裝 Docker Engine
# 卸載舊版本 Docker(如有) sudoapt-get remove docker.io docker-doc containerd # 添加 Docker 官方源 curl -fsSL https://download.docker.com/linux/ubuntu/gpg |sudogpg --dearmor -o /etc/apt/keyrings/docker.gpg echo"deb [arch=$(dpkg --print-architecture)signed-by=/etc/apt/keyrings/docker.gpg] https://download.docker.com/linux/ubuntu$(lsb_release -cs)stable"|sudotee/etc/apt/sources.list.d/docker.list > /dev/null sudoapt-get update sudoapt-get install docker-ce docker-ce-cli containerd.io -y # 驗證 Docker docker info root@i-28e6iose:/home/ubuntu# docker info Client: Docker Engine - Community Version: 28.0.1 Context: default Debug Mode:false Plugins: buildx: Docker Buildx (Docker Inc.) Version: v0.21.1 Path: /usr/libexec/docker/cli-plugins/docker-buildx compose: Docker Compose (Docker Inc.) Version: v2.33.1 Path: /usr/libexec/docker/cli-plugins/docker-compose Server: Containers: 10 Running: 10 Paused: 0 Stopped: 0 Images: 10 Server Version: 28.0.1
3.配置 NVIDIA Container Toolkit
在這里插入圖片描述 nvidia/cuda:12.2.0-base-ubuntu22.04是一個基于 Ubuntu 22.04 操作系統(tǒng)的 NVIDIA CUDA 基礎(chǔ)鏡像,用于運行需要GPU 加速的計算環(huán)境。CUDA(Compute Unified Device Architecture)是 NVIDIA 提供的一個并行計算平臺和編程模型,它使開發(fā)者可以使用 NVIDIA GPU 進行高性能計算。這個鏡像提供了 CUDA 12.2.0 版本,適用于需要利用 NVIDIA GPU 進行深度學習、科學計算和其他計算密集型任務的場景。CUDA 12.2.0 版本帶來了許多改進和新特性,包括對新架構(gòu)的支持、性能優(yōu)化和新 APIhttps://docs.nvidia.com/datacenter/cloud-native/container-toolkit/latest/install-guide.html
# 添加倉庫并安裝工具包 curl -fsSL https://nvidia.github.io/libnvidia-container/gpgkey |sudogpg --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 |sudotee/etc/apt/sources.list.d/nvidia-container-toolkit.list sudoapt-get update sudoapt-get install nvidia-container-toolkit -y # 配置 Docker 運行時 sudonvidia-ctk runtime configure --runtime=docker sudosystemctl restart docker root@i-28e6iose:/home/ubuntu#sudonvidia-ctk runtime configure --runtime=docker INFO[0000] Loading config from /etc/docker/daemon.json INFO[0000] Wrote updated config to /etc/docker/daemon.json INFO[0000] It is recommended that docker daemon be restarted. # 驗證 CUDA 容器 docker run --rm--gpus all nvidia/cuda:12.2.0-base-ubuntu22.04 nvidia-smi root@i-28e6iose:/home/ubuntu# docker run --rm--gpus all nvidia/cuda:12.2.0-base-ubuntu22.04 nvidia-smi Sun Mar 9 1355 2025 +-----------------------------------------------------------------------------------------+ | NVIDIA-SMI 570.124.06 Driver Version: 570.124.06 CUDA Version: 12.8 | |-----------------------------------------+------------------------+----------------------+ | GPU Name Persistence-M | Bus-Id Disp.A | Volatile Uncorr. ECC | | Fan Temp Perf Pwr:Usage/Cap | Memory-Usage | GPU-Util Compute M. | | | | MIG M. | |=========================================+========================+======================| | 0 NVIDIA GeForce RTX 2080 Ti Off | 000000000C.0 Off | N/A | | 22% 26C P8 21W / 250W | 4MiB / 11264MiB | 0% Default | | | | N/A | +-----------------------------------------+------------------------+----------------------+ +-----------------------------------------------------------------------------------------+ | Processes: | | GPU GI CI PID Type Process name GPU Memory | | ID ID Usage | |=========================================================================================| | No running processes found | +-----------------------------------------------------------------------------------------+
三、部署 GPUStack 容器
1.運行 GPUStack 主節(jié)點
docker run -d --gpus all -p 890:80 --ipc=host --name gpustack -v gpustack-data:/var/lib/gpustack gpustack/gpustack:latest
?參數(shù)說明
?--gpus all:啟用所有 GPU 資源。
?--ipc=host:共享主機 IPC 命名空間,提升性能。
?-v gpustack-data:持久化存儲配置與模型數(shù)據(jù)。
2.獲取初始密碼
dockerexec-it gpustackcat/var/lib/gpustack/initial_admin_password root@i-28e6iose:/home/ubuntu# dockerexec-it gpustackcat/var/lib/gpustack/initial_admin_password rjl@Ainm3dtQ #賬戶信息: admin/rjl@Ainm3dtQ #修改密碼:P@88w0rd訪問http://<服務器IP>,使用用戶名admin和上述密碼登錄,首次需重置密碼。
在這里插入圖片描述
四、擴展 GPU 集群
1.添加 Worker 節(jié)點
? 在主節(jié)點獲取 Token:
dockerexec-it gpustackcat/var/lib/gpustack/token
? 在 Worker 節(jié)點運行:
docker run -d --gpus all --network=host --ipc=host gpustack/gpustack --server-url http://<主節(jié)點IP> --token <獲取的Token>
五、功能使用示例
1.部署大模型
在 GPUStack 控制臺的Models頁面,支持從 Hugging Face 或本地導入模型。例如部署 Llama3.2 模型時,系統(tǒng)會自動分配 GPU 資源并生成 API 端點。
2.Playground 調(diào)測
在 Playground 中可測試多模態(tài)模型(如 Stable Diffusion)、文本嵌入模型(BERT)等,支持多模型對比與參數(shù)優(yōu)化。
六、常見問題
?GPU 未識別:檢查nvidia-smi是否正常,并確認 Docker 運行時配置正確。
?容器啟動失敗:確保已啟用--ipc=host并掛載持久化卷。
?網(wǎng)絡問題:跨節(jié)點通信需開放防火墻的 80 端口及內(nèi)部 RPC 端口(默認為 6789)。
七、參考資源
?GPUStack 官方 Docker 部署文檔
?NVIDIA Container Toolkit 配置指南
通過以上步驟,您可快速搭建一個支持異構(gòu) GPU 資源調(diào)度的企業(yè)級大模型服務平臺,實現(xiàn)從單機到集群的高效擴展。
鏈接:https://blog.csdn.net/shenhonglei1234/article/details/146139022
-
gpu
+關(guān)注
關(guān)注
28文章
4949瀏覽量
131274 -
開源
+關(guān)注
關(guān)注
3文章
3690瀏覽量
43840 -
模型
+關(guān)注
關(guān)注
1文章
3521瀏覽量
50441 -
Docker
+關(guān)注
關(guān)注
0文章
515瀏覽量
12974
原文標題:告別環(huán)境折磨!Docker+GPUStack一鍵部署教程,NVIDIA顯卡秒變AI工作站
文章出處:【微信號:magedu-Linux,微信公眾號:馬哥Linux運維】歡迎添加關(guān)注!文章轉(zhuǎn)載請注明出處。
發(fā)布評論請先 登錄
docker運行datax實現(xiàn)數(shù)據(jù)同步方案
在Ubuntu 18.04 for Arm上運行的TensorFlow和PyTorch的Docker映像
Docker鏡像的詳細講解
開源的Docker中運行Mac OS是什么體驗?
如何在Docker容器中運行Nginx
Docker 教程:如何將Helix QAC作為容器創(chuàng)建并運行

如何在macOS系統(tǒng)中用Docker運行macOS鏡像呢?

docker exec命令的使用方法
linux關(guān)閉docker的命令
如何啟動本機docker服務
docker進入容器的方法有哪些
docker核心組件有哪些
【鴻蒙】OpenHarmony運行docker詳細步驟

評論