26 lines
640 B
YAML
26 lines
640 B
YAML
services:
|
|
fcb_ai_dev:
|
|
image: fcb_ai_dev:0.0.1
|
|
build:
|
|
context: .
|
|
dockerfile: Dockerfile.dev
|
|
# 容器名称，可自定义
|
|
container_name: fcb_ai_dev
|
|
hostname: fcb_ai_dev
|
|
# 始终重启容器
|
|
restart: always
|
|
# 端口映射，根据需要修改主机端口
|
|
ports:
|
|
- "8022:22"
|
|
# 数据卷映射，根据实际路径修改
|
|
volumes:
|
|
- ./log:/app/log
|
|
- ./model:/app/model
|
|
# 启用GPU支持
|
|
deploy:
|
|
resources:
|
|
reservations:
|
|
devices:
|
|
- device_ids: [ '0', '1' ]
|
|
capabilities: [ 'gpu' ]
|
|
driver: 'nvidia' |