services:
  ollama:
    image: ollama/ollama
    container_name: ollama
    ports:
      - "11434:11434" # 主机端口映射到容器端口
    volumes:
      - ollama:/root/.ollama # 挂载数据卷
    deploy:
      resources:
        reservations:
          devices:
            - driver: nvidia # 指定使用 nvidia GPU 驱动
              count: all # 使用所有可用 GPU
              capabilities: [gpu] # 明确使用 GPU 的计算功能
    restart: unless-stopped # 设置容器异常停止时自动重启

volumes:
  ollama: