Rocm7.2已经出了但是不兼容AI-MAX 395，下面是安装Rocm7.1.1和对应的rocm/pytorch
apt update
apt install vim wget gpg curl git -y
apt upgrade -y
wget https://repo.radeon.com/amdgpu-install/7.1.1/ubuntu/noble/amdgpu-install_7.1.1.70101-1_all.deb
apt install ./amdgpu-install_7.1.1.70101-1_all.deb
apt update
apt install python3-setuptools python3-wheel libjpeg-dev python3-dev python3-pip
groupadd render
usermod -a -G render,video ubuntu
apt install rocm
apt update
apt install “linux-headers-$(uname -r)” “linux-modules-extra-$(uname -r)”
apt install amdgpu-dkms
docker pull rocm/pytorch:rocm7.1.1_ubuntu24.04_py3.12_pytorch_release_2.9.1