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分钟左右,效果也还可以接受。
浙公网安备 33010602011771号