Skip to content

Commit

Permalink
Merge branch 'main' into features/kunlunxin_glm
Browse files Browse the repository at this point in the history
  • Loading branch information
dynamicheart authored Nov 3, 2023
2 parents 3fcad42 + c5058d9 commit 5c494f9
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions training/kunlunxin/docker_image/pytorch/pytorch_install.sh
Original file line number Diff line number Diff line change
Expand Up @@ -5,8 +5,8 @@ set -xe
pip install https://bd.bcebos.com/klx-pytorch-ipipe-bd/flagperf/latest/xacc-0.1.0-cp38-cp38-linux_x86_64.whl
pip install https://bd.bcebos.com/klx-pytorch-ipipe-bd/flagperf/latest/xmlir-0.0.1-cp38-cp38-linux_x86_64.whl

pip install psutil==5.9.5 -i https://pypi.tuna.tsinghua.edu.cn/simple
pip install accelerate==0.20.3 -i https://pypi.tuna.tsinghua.edu.cn/simple
pip install tabulate==0.9.0 -i https://pypi.tuna.tsinghua.edu.cn/simple
pip install psutil==5.9.5 -i https://pypi.tuna.tsinghua.edu.cn/simple
pip install accelerate==0.20.3 -i https://pypi.tuna.tsinghua.edu.cn/simple
pip install tabulate==0.9.0 -i https://pypi.tuna.tsinghua.edu.cn/simple

python -m xacc.install

0 comments on commit 5c494f9

Please sign in to comment.