21 lines
699 B
YAML
21 lines
699 B
YAML
|
version: '3'
|
||
|
|
services:
|
||
|
|
app:
|
||
|
|
image: aiflask:latest
|
||
|
|
container_name: aiflask_container
|
||
|
|
ports:
|
||
|
|
- "5000:5000"
|
||
|
|
volumes:
|
||
|
|
- /home/ooin/lzz/AI_train/AIData/Datasets:/home/AIData/Datasets # 将本地目录挂载到容器的 /data 目录
|
||
|
|
- /home/ooin/lzz/AI_train/AIData/model_factory:/home/AIData/model_factory # 将本地目录挂载到容器的 /data 目录
|
||
|
|
runtime: nvidia # 启用 GPU 支持
|
||
|
|
environment:
|
||
|
|
PYTHONPATH: /home/lzz/aiflask
|
||
|
|
deploy:
|
||
|
|
resources:
|
||
|
|
reservations:
|
||
|
|
devices:
|
||
|
|
- capabilities: [gpu]
|
||
|
|
command: ["/opt/anaconda3/envs/AIFlask/bin/gunicorn", "--bind", "0.0.0.0:5000", "home.lzz.aiflask.app:app"]
|
||
|
|
|