heygem本地部署遇到的两个坑

我的电脑显卡是nivdia3060,想试一下开源的数字人heygem,使用过程中遇到如下问题。

1、docker的安装
启动或关闭windows功能中,有虚拟机平台和适用于linux的windows子系统需要打开,如果有问题可能需要先关闭再打开。才能安装wsl
主板和系统的相关虚拟化选项都打开了,还是不能安装docker,后来发现还需要运行一个命令。执行完之后可以正常运行dockerl

C:\WINDOWS\system32>bcdedit.exe /enum | findstr -i hyper
hypervisorlaunchtype    Off

C:\WINDOWS\system32>bcdedit.exe /set hypervisorlaunchtype Auto
操作成功完成。

C:\WINDOWS\system32>bcdedit.exe /enum | findstr -i hyper
hypervisorlaunchtype    Auto

2、代码克隆到本地之后,运行会报显卡的错误,本地的显卡驱动已经装过了,后来发现有一个容器并不报错,只有主容器报错。
检查 HeyGem.ai/deploy/docker-compose.yml 文件,发现device节点不同,修改后可以正常使用了。

networks:
  ai_network:
    driver: bridge

services:
  heygem-tts:
    image: guiji2025/fish-speech-ziming
    container_name: heygem-tts
    restart: always
    runtime: nvidia
    environment:
      - NVIDIA_VISIBLE_DEVICES=0
      - NVIDIA_DRIVER_CAPABILITIES=compute,graphics,utility,video,display
    ports:
      - '18180:8080'
    volumes:
      - d:/heygem_data/voice/data:/code/data
    command: /bin/bash -c "/opt/conda/envs/python310/bin/python3 tools/api_server.py --listen 0.0.0.0:8080"
    networks:
      - ai_network
  heygem-asr:
    image: guiji2025/fun-asr
    container_name: heygem-asr
    restart: always
    runtime: nvidia
    privileged: true
    working_dir: /workspace/FunASR/runtime
    ports:
      - '10095:10095'
    command: sh /run.sh
    deploy:
      resources:
        reservations:
          devices:
            - driver: nvidia
              count: all
              capabilities: [gpu]
    networks:
      - ai_network
  heygem-gen-video:
    image: guiji2025/heygem.ai
    container_name: heygem-gen-video
    restart: always
    runtime: nvidia
    privileged: true
    volumes:
      - d:/heygem_data/face2face:/code/data
    environment:
      - PYTORCH_CUDA_ALLOC_CONF=max_split_size_mb:512
    deploy:
      resources:
        reservations:
          devices:
            - capabilities: [gpu]
    shm_size: '8g'
    ports:
      - '8383:8383'
    command: python /code/app_local.py
    networks:
      - ai_network

主要是这里:原始文件没有前边两行,增加后可以使用了

devices:
  - driver: nvidia 
    count: all  
    capabilities: [gpu]

3060显卡,生成一分钟数字人视频,用时大概2分钟左右,效果也还可以接受。

posted @ 2025-05-12 15:54  sec001  阅读(273)  评论(0)    收藏  举报