|
|
@@ -1,14 +1,29 @@
|
|
|
version: '3.8'
|
|
|
|
|
|
services:
|
|
|
- app:
|
|
|
+ triton-server:
|
|
|
build:
|
|
|
context: .
|
|
|
dockerfile: Dockerfile
|
|
|
- container_name: task-server-app
|
|
|
+ image: my-tritonserver:latest # 可自定义构建后的镜像名称
|
|
|
ports:
|
|
|
- "8000:8000"
|
|
|
- volumes:
|
|
|
- - .:/app
|
|
|
environment:
|
|
|
- - PYTHONUNBUFFERED=1
|
|
|
+ - TZ=Asia/Shanghai
|
|
|
+ - PYTHONDONTWRITEBYTECODE=1
|
|
|
+ - PYTHONUNBUFFERED=1
|
|
|
+ - PIP_DISABLE_PIP_VERSION_CHECK=on
|
|
|
+ # 如果需要GPU支持，取消下面的注释
|
|
|
+ # deploy:
|
|
|
+ # resources:
|
|
|
+ # reservations:
|
|
|
+ # devices:
|
|
|
+ # - driver: nvidia
|
|
|
+ # count: 1
|
|
|
+ # capabilities: [gpu]
|
|
|
+ # 如果需要挂载模型数据卷，取消下面的注释
|
|
|
+ # volumes:
|
|
|
+ # - ./models:/app/models
|
|
|
+ # 如果需要设置资源限制，取消下面的注释
|
|
|
+ # mem_limit: 8g
|
|
|
+ # cpus: 4.0
|
luojunhui