ubuntu20.04部署ROCm深度学习

按照官网给的教程一步步来
官网链接
担心有的人每仔细看官网而错误安装环境,我把命令行都写下来,大家直接复制粘贴到命令台就好了

运行以下代码以确保系统是最新的:

sudo apt update

sudo apt dist-upgrade

sudo apt install libnuma-dev

sudo reboot

添加ROCm apt存储库:

sudo apt install wget gnupg2

wget -q -O - https://repo.radeon.com/rocm/rocm.gpg.key | sudo apt-key add -

echo 'deb [arch=amd64] https://repo.radeon.com/rocm/apt/4.3/ ubuntu main' | sudo tee /etc/apt/sources.list.d/rocm.list

gpg键可能会改变;确保在安装新版本时对其进行更新。如果密钥签名验证在更新时失败,请从ROCm apt存储库重新添加密钥。

wget -q -O - https://repo.radeon.com/rocm/rocm.gpg.key | sudo apt-key add -

安装ROCm元软件包。更新相应的存储库列表并安装rocm dkms元软件包:

sudo apt update

sudo apt install rocm-dkms && sudo reboot

重新启动系统。

重新启动系统后,运行以下命令以验证ROCm安装是否成功。如果两个命令都列出了GPU,则认为安装成功。

/opt/rocm/bin/rocminfo
/opt/rocm/opencl/bin/clinfo

你可能感兴趣的:(ubuntu)