从源码构建zentorch v2.11.0.2:Llama-3.3-70B-Instruct-da8w8-torchao-v0.17.0依赖环境搭建
从源码构建zentorch v2.11.0.2Llama-3.3-70B-Instruct-da8w8-torchao-v0.17.0依赖环境搭建【免费下载链接】Llama-3.3-70B-Instruct-da8w8-torchao-v0.17.0项目地址: https://ai.gitcode.com/hf_mirrors/amd/Llama-3.3-70B-Instruct-da8w8-torchao-v0.17.0Llama-3.3-70B-Instruct-da8w8-torchao-v0.17.0是基于Meta Llama 3.3-70B-Instruct模型优化的8位量化版本专为AMD EPYC CPU推理设计。本文将详细介绍如何从源码构建zentorch v2.11.0.2及其依赖环境确保与该模型完美兼容。核心依赖版本锁定成功搭建环境的关键在于严格匹配以下版本组合PyTorch: 2.11.0cpuTorchAO: 0.17.0量化框架ZenDNN: v6.0.0AMD深度学习加速库zentorch: v2.11.0.2必须从源码构建vLLM: 0.23.0推理引擎⚠️版本兼容性警告该模型通过TorchAO v0.17.0量化仅兼容PyTorch v2.11.0和ZenDNN v6.0.0其他版本会导致加载失败。环境准备步骤1. 基础系统要求操作系统Linux推荐Ubuntu 20.04或CentOS 8硬件AMD EPYC处理器支持AVX2指令集工具链GitPython 3.9-3.11Conda或venv虚拟环境GCC 9.4.0源码编译需要2. 克隆项目仓库git clone https://gitcode.com/hf_mirrors/amd/Llama-3.3-70B-Instruct-da8w8-torchao-v0.17.0 cd Llama-3.3-70B-Instruct-da8w8-torchao-v0.17.03. 创建并激活虚拟环境# 使用conda创建环境 conda create -n zentorch-env python3.10 -y conda activate zentorch-env # 或使用venv python -m venv zentorch-env source zentorch-env/bin/activate # Linux/Mac4. 安装核心依赖通过pip安装指定版本的PyTorch、TorchAO和vLLMpip install --extra-index-url https://download.pytorch.org/whl/cpu \ --extra-index-url https://wheels.vllm.ai/cpu/ \ torch2.11.0cpu \ vllm0.23.0 \ torchao0.17.0 \ lm-eval[vllm]0.4.12 \ huggingface_hub5. 安装CPU运行时库conda install -c conda-forge gperftools2.17.2 llvm-openmp18.1.8 --no-deps -y从源码构建zentorch v2.11.0.21. 获取ZenDNN PyTorch插件源码git clone https://github.com/amd/ZenDNN-pytorch-plugin.git cd ZenDNN-pytorch-plugin git checkout v2.11.0.2 # 切换到指定版本2. 编译安装zentorch# 设置编译参数 export ZENDNN_ROOT/path/to/your/zendnn export CMAKE_PREFIX_PATH$(python -c import torch; print(torch.utils.cmake_prefix_path)) # 编译并安装 python setup.py install提示若编译过程中遇到依赖问题参考项目根目录下的README.md文件中的从源码构建章节。环境变量配置设置必要的环境变量以启用优化和指定运行时库路径# TorchInductor zentorch优化 export TORCHINDUCTOR_FREEZING1 export TORCHINDUCTOR_AUTOGRAD_CACHE0 export VLLM_USE_AOT_COMPILE0 export ZENDNNL_MATMUL_ALGO1 # 指定CPU运行时库路径 export LD_PRELOAD/path/to/lib/libtcmalloc_minimal.so.4:/path/to/lib/libiomp5.so${LD_PRELOAD::$LD_PRELOAD}使用以下命令定位库文件路径find / -name libtcmalloc_minimal.so.4 find / -name libiomp5.so验证环境正确性1. 检查版本信息python -c import torch; print(PyTorch version:, torch.__version__) python -c import torchao; print(TorchAO version:, torchao.__version__)预期输出PyTorch version: 2.11.0cpu TorchAO version: 0.17.02. 运行模型推理测试import torch from transformers import AutoModelForCausalLM, AutoTokenizer model_id ./ # 当前项目目录 tokenizer AutoTokenizer.from_pretrained(model_id) model AutoModelForCausalLM.from_pretrained( model_id, dtypetorch.bfloat16, device_mapcpu, trust_remote_codeTrue ) inputs tokenizer(What is AMD ZenDNN?, return_tensorspt) outputs model.generate(**inputs, max_new_tokens50) print(tokenizer.decode(outputs[0], skip_special_tokensTrue))常见问题解决编译错误确保GCC版本≥9.4.0缺少依赖时安装build-essential库文件找不到使用ldconfig更新动态链接库缓存版本冲突通过pip list | grep torch检查是否存在多个PyTorch版本总结通过本文步骤您已成功搭建Llama-3.3-70B-Instruct-da8w8-torchao-v0.17.0模型的依赖环境包括从源码构建zentorch v2.11.0.2。关键在于严格遵循版本要求和环境变量配置以确保AMD CPU推理性能优化。详细量化实现可参考项目中的量化示例代码更多高级配置请查阅官方文档。【免费下载链接】Llama-3.3-70B-Instruct-da8w8-torchao-v0.17.0项目地址: https://ai.gitcode.com/hf_mirrors/amd/Llama-3.3-70B-Instruct-da8w8-torchao-v0.17.0创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考