version: '3.9' | |
services: | |
biencoder-app: | |
build: | |
context: . | |
dockerfile: Dockerfile.biencoder | |
container_name: biencoder_docker | |
runtime: nvidia | |
environment: | |
- NVIDIA_VISIBLE_DEVICES=all | |
- PYTHONUNBUFFERED=1 | |
volumes: | |
- .:/app | |
working_dir: /app | |
ports: | |
- "8000:8000" | |