26 lines
909 B
YAML
Executable File
26 lines
909 B
YAML
Executable File
# GPU 叠加配置（Compose V2：docker compose）
|
||
# 用法：docker compose -f docker-compose.yml -f docker-compose.gpu-v2.yml up -d
|
||
#
|
||
# 前提：主机已安装 nvidia-container-toolkit
|
||
# distribution=$(. /etc/os-release; echo $ID$VERSION_ID)
|
||
# curl -s -L https://nvidia.github.io/libnvidia-container/gpgkey | sudo apt-key add -
|
||
# curl -s -L https://nvidia.github.io/libnvidia-container/$distribution/libnvidia-container.list \
|
||
# | sudo tee /etc/apt/sources.list.d/nvidia-container-toolkit.list
|
||
# sudo apt-get update && sudo apt-get install -y nvidia-container-toolkit
|
||
# sudo nvidia-ctk runtime configure --runtime=docker
|
||
# sudo systemctl restart docker
|
||
|
||
version: '3.8'
|
||
|
||
services:
|
||
app:
|
||
deploy:
|
||
resources:
|
||
reservations:
|
||
devices:
|
||
- driver: nvidia
|
||
count: all
|
||
capabilities: [gpu]
|
||
environment:
|
||
- ASSEMBLY_USE_GPU=1
|