8000 GitHub - IAMJOYBO/ktransformers: 清华大学 KTransformers Docker Image Build Tool
[go: up one dir, main page]
More Web Proxy on the site http://driver.im/
Skip to content

IAMJOYBO/ktransformers

Repository files navigation

清华大学 KTransformers Docker Image Build Tool

Docker镜像

registry.cn-hangzhou.aliyuncs.com/joybo/ktransformers

序号 Tag 构建时间 指令集 PyTorch版本 New WEB 架构
01 v2025.04.12-action 2025.4.12 NATIVE 2.6.0+cu126 最新镜像 amd64
02 v2025.04.13-action-web 2025.4.13 NATIVE 2.6.0+cu126 最新镜像 amd64

PyTorch基础镜像:https://hub.docker.com/r/pytorch/pytorch/tags

Docker Compose 示例(无WEB)

services:
  ktransformers:
    image: registry.cn-hangzhou.aliyuncs.com/joybo/ktransformers:v2025.04.12-action
    container_name: ktransformers
    hostname: ktransformers
    environment:
      - TZ=Asia/Shanghai
      # - NVIDIA_VISIBLE_DEVICES=0
    volumes:
      - ./DeepSeek-V2-Lite-Chat-GGUF:/app/model/DeepSeek-V2-Lite-Chat-GGUF
    ports:
      - "10002:10002"
    runtime: nvidia
    deploy:
      resources:
        reservations:
          devices:
          - driver: nvidia
            # device_ids: ['0']
            capabilities: [gpu]
    stdin_open: true
    tty: true
    restart: no
    networks:
      - ktransformers
    entrypoint: ["python", "-m", "ktransformers.local_chat", "--model_path=/app/model/DeepSeek-V2-Lite-Chat", "--gguf_path=/app/model/DeepSeek-V2-Lite-Chat-GGUF"]

networks:
  ktransformers:
    driver: bridge
    name: ktransformers

Docker Compose 示例(带WEB)