apt update
apt upgrade -y
apt install wget sudo git
wget -q -O - http://repo.radeon.com/rocm/rocm.gpg.key | apt-key add -
apt update
apt install -y rocm-dev rocm-utils rocm-libs miopen-hip miopengemm rccl
export PATH=/opt/rocm-3.5.1/bin:$PATH
# python 3.6
pip install wheel
python3 setup.py bdist_wheel

echo "gfx803" >> $ROCM_PATH/bin/target.lst'