File size: 972 Bytes
7e91bd7 671defb 7e91bd7 671defb 7e91bd7 671defb 7e91bd7 671defb |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 |
#!/bin/bash -
#===============================================================================
#
# 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
|