DeepSeek大模型指定github项目版本安装环境
最近DeepSeek非常的火爆,有一些公司复现了DeepSeek,如open-r1, 但其依赖的环境往往是最新的,甚至是新增的功能,整个生态安装没有完善。需要需要指定特定的依赖安装:
查看open-r1的setup.py发现,lighteval,trl和transformers是需要源码安装,甚至还需要从某个commit中安装:
# 省略部分
_deps = [
"accelerate>=1.2.1",
"lighteval @ git+https://github.com/huggingface/lighteval.git@0e462692436e1f0575bdb4c6ef63453ad9bde7d4#egg=lighteval[math]",
"torch>=2.5.1",
"transformers @ git+https://github.com/huggingface/transformers.git@main",
"trl @ git+https://github.com/huggingface/trl.git@main",
"vllm>=0.7.0",
"wandb>=0.19.1",
]
安装时执行
pip install -e "."
安装过程中需要从github中下载代码,但下载过程中碰到如下错误:
原因分析:
git拉取代码默认是不支持拉取不信任的代码的,因此需要配置如下环境变量以便信任该代码:
export GIT_CLONE_PROTECTION_ACTIVE=false
重新执行安装命令即可顺利下载git代码。