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

最后

以上就是儒雅小松鼠为你收集整理的ubuntu20.04部署ROCm深度学习的全部内容，希望文章能够帮你解决ubuntu20.04部署ROCm深度学习所遇到的程序开发问题。

如果觉得靠谱客网站的内容还不错，欢迎将靠谱客网站推荐给程序员好友。