| #=============================================================================== | |
| # | |
| # FILE: rocm52.py38.ds.sh | |
| # | |
| # USAGE: ./rocm52.py38.ds.sh | |
| # | |
| # DESCRIPTION: | |
| # | |
| # OPTIONS: --- | |
| # REQUIREMENTS: --- | |
| # BUGS: --- | |
| # NOTES: --- | |
| # AUTHOR: fuliucansheng (fuliu), fuliucansheng@gmail.com | |
| # ORGANIZATION: | |
| # CREATED: 06/17/2023 00:14 | |
| # REVISION: --- | |
| #=============================================================================== | |
| set -o nounset # Treat unset variables as an error | |
| sudo apt update | |
| sudo apt install -y python3.8-dev python3-pip | |
| sudo apt install -y ninja-build | |
| curl https://bootstrap.pypa.io/get-pip.py -o get-pip.py | |
| sudo python3.8 get-pip.py | |
| sudo rm -rf get-pip.py | |
| sudo pip3.8 install torch==1.13.1+rocm5.2 torchvision==0.14.1+rocm5.2 torchaudio==0.13.1 --extra-index-url https://download.pytorch.org/whl/rocm5.2 | |
| sudo pip3.8 install deepspeed ninja | |