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
  • 2
    点赞
  • 18
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值