Skip to content

Commit

Permalink
update env install (#935)
Browse files Browse the repository at this point in the history
Co-authored-by: nifeng <[email protected]>
  • Loading branch information
luyao-cv and nemonameless authored Dec 31, 2024
1 parent 8cad64b commit b5248ae
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions build_env.sh
Original file line number Diff line number Diff line change
Expand Up @@ -21,16 +21,16 @@ echo "开始安装 PaddleMIX 及其依赖..."

# 安装 PaddleMIX
echo "安装 PaddleMIX..."
pip install -e .
pip install -e . --user

# 安装 ppdiffusers
echo "安装 ppdiffusers..."
cd ppdiffusers
pip install -e .
pip install -e . --user
cd ..
#注:ppdiffusers部分模型需要依赖 CUDA 11.2 及以上版本,如果本地机器不符合要求,建议前往 [AI Studio](https://aistudio.baidu.com/index) 进行模型训练、推理任务。
#如果希望使用**bf16**训练推理,请使用支持**bf16**的GPU,如A100。

# 安装依赖包
echo "安装依赖包..."
pip install -r requirements.txt
pip install -r requirements.txt --user

0 comments on commit b5248ae

Please sign in to comment.