update docker files

1. add docker-npu (Dockerfile and docker-compose.yml)
  2. move cuda docker to docker-cuda and tiny changes to adapt to the new path


Former-commit-id: 5431c1f18aadb072208efe7fd8e36fdcfbf807c2
This commit is contained in:
MengqingCao
2024-06-24 10:57:36 +00:00
parent 826d7808b4
commit af2607de1a
6 changed files with 187 additions and 19 deletions

View File

@@ -0,0 +1,30 @@
services:
llamafactory:
build:
dockerfile: Dockerfile
context: .
args:
INSTALL_BNB: false
INSTALL_VLLM: false
INSTALL_DEEPSPEED: false
PIP_INDEX: https://pypi.org/simple
container_name: llamafactory
volumes:
- ../../hf_cache:/root/.cache/huggingface/
- ../../data:/app/LLaMA-Factory/data
- ../../output:/app/LLaMA-Factory/output
ports:
- "7860:7860"
- "8000:8000"
ipc: host
tty: true
stdin_open: true
command: bash
deploy:
resources:
reservations:
devices:
- driver: nvidia
count: "all"
capabilities: [gpu]
restart: unless-stopped