Initial commit

docker-compose.gpu-v2.yml

@@ -0,0 +1,25 @@
+# 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