22 lines
546 B
YAML
22 lines
546 B
YAML
services:
|
|
deepseek_ocr_vllm:
|
|
build:
|
|
context: .
|
|
dockerfile: Dockerfile
|
|
image: deepseek-ocr-vllm:cu126
|
|
container_name: deepseek_ocr_vllm
|
|
working_dir: /workspace
|
|
volumes:
|
|
- ./:/workspace
|
|
gpus: all
|
|
shm_size: "8g"
|
|
ipc: "host"
|
|
environment:
|
|
- HF_HOME=/workspace/.cache/huggingface
|
|
- CUDA_HOME=/usr/local/cuda
|
|
- LD_LIBRARY_PATH=/usr/local/cuda/lib64:${LD_LIBRARY_PATH}
|
|
- PIP_DISABLE_PIP_VERSION_CHECK=1
|
|
- PYTHONUNBUFFERED=1
|
|
tty: true
|
|
entrypoint: ["/bin/bash"]
|