#=============================================================================== | |
# | |
# FILE: py38.sh | |
# | |
# USAGE: ./py38.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 python3.8-dev python3-pip | |
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 | |