10秒变3D!腾讯混元Hunyuan3D-1.0,让图片和文字活起来
今天给大家带来一个超给力的3D技术—腾讯的混元Hunyuan3D-1.0,一个能把图片和文字变成3D模型的神奇工具!”
“这个技术牛在哪儿呢?它用了一个叫做多视角扩散模型的黑科技,能从不同角度看一个东西,让3D模型更真实、更详细。”
“而且,速度超快!轻量版10秒,标准版25秒,就能搞定一个3D模型!
社区镜像
觉得麻烦的直接用这个镜像,已安装所需依赖
测试用的硬件配置:
-
系统: Ubuntu
-
GPU 名称: NVIDIA GeForce RTX 4090 D
-
系 统 盘/ :30G
-
数 据 盘/root/autodl-tmp:50G
-
内存:60 GB
软件:
-
Python 版本: 3.10
-
PyTorch 版本: 2.1.2+cu121
-
CUDA 版本: 12.1
#开启学术加速
source /etc/network_turbo
#取消学术加速
unset http_proxy && unset https_proxy
#查看剩余空间
source /etc/network_turbo
conda init
#激活当前环境
conda activate /root/miniconda3
安装pytorch3d
说明文档:https://github.com/facebookresearch/pytorch3d/blob/main/INSTALL.md
「pytorch3d-0.7.5-py310_cu121_pyt210-linux_x86_64.whl」
pytorch3d预编译whl
pip install pytorch3d-0.7.5-py310_cu121_pyt210-linux_x86_64.whl
#检查cuda可用性
python -c "import torch; print(f'PyTorch 版本: {torch.__version__}'); print(f'CUDA 是否可用: {torch.cuda.is_available()}'); print(f'CUDA 版本: {torch.version.cuda if torch.cuda.is_available() else "N/A"}'); print(f'GPU 数量: {torch.cuda.device_count()}'); print(f'GPU 名称: {torch.cuda.get_device_name(0) if torch.cuda.is_available() else "N/A"}')"
#返回示例
PyTorch 版本: 2.1.2+cu121 CUDA 是否可用: True CUDA 版本: 12.1 GPU 数量: 1 GPU 名称: NVIDIA GeForce RTX 4090 D
#检查cuda驱动版本
nvcc --version
#返回示例 nvcc: NVIDIA (R) Cuda compiler driver Copyright (c) 2005-2023 NVIDIA Corporation Built on Mon_Apr__3_17:16:06_PDT_2023 Cuda compilation tools, release 12.1, V12.1.105 Build cuda_12.1.r12.1/compiler.32688072_0
项目部署过程:
#由于模型文件约28G 需在数据盘部署
cd /root/autodl-tmp
git clone https://github.com/Tencent/Hunyuan3D-1
cd Hunyuan3D-1/
#删除 pip install git+https://github.com/facebookresearch/pytorch3d.git
vim env_install.sh
#删除整行
dd
#保存退出
Esc :wq
bash env_install.sh pip install tbb
下载模型
我用夸克网盘分享了「腾讯混元Hunyuan3D-1.0模型文件weights.7z」,点击链接即可保存。
链接:https://pan.quark.cn/s/bf4ba82fb483
pip install "huggingface_hub[cli]"
#设置镜像加速
export HF_ENDPOINT=https://hf-mirror.com
mkdir weights\hunyuanDiT
huggingface-cli download tencent/Hunyuan3D-1 --local-dir weights
huggingface-cli download Tencent-Hunyuan/HunyuanDiT-v1.1-Diffusers-Distilled --local-dir weights/hunyuanDiT
运行
单显卡小于30GB请使用 –use_lite –save_memory
python3 app.py --use_lite --save_memory
显卡大于30GB使用
python3 app.py
-
推理Std 标准版需要 30GB VRAM,4090单卡是24GB,如使用标准版推理需要租用2张显卡
-
推理 Lite版需要 22GB VRAM(18G VRAM 和 –save_memory) 。
-
注意:–save_memory 会增加推理时间
© 版权声明
文章版权归作者所有,未经允许请勿转载。
THE END
暂无评论内容