本人买的AMD显卡,最近要在ubuntu环境做python深度学习,所以必须折腾一番,结果发现还不复杂。本人的系统是Ubuntu 20.04.6,显卡是RX6600,亲测有效。
(0) 如果已经配置了显卡驱动但是没有成功,或配置成功想更换版本,则先卸载原有的驱动,重新开始:
~$ sudo amdgpu-install --uninstall
(1) 在AMD官网下载需要的驱动包:
驱动包链接
得到amdgpu-install_xxx_all.deb(xxx是版本号)
(2) cd到驱动包路径下,安装驱动包:
~$ sudoaptinstall ./amdgpu-install_xxx_all.deb
(3) 更新软件:
~$ sudoapt update
~$ sudoapt upgrade -y
(4) 安装驱动:
~$ sudo amdgpu-install --no-dkms
~$ sudoaptinstall rocm-dev
~$ sudoreboot# 等待重启
(5)配置环境:
~$ ls-l /dev/dri/render*
~$ sudousermod-a-G render $LOGNAME
~$ sudousermod-a-G video $LOGNAME# $LOGNAME不需要修改
~$ sudoreboot# 等待重启
(6) 理论上如果一路上没有报错,便配置成功了,可以输入下列命令验证
~$ rocm-smi
# 监控GPU性能
~$ /opt/rocm/bin/rocminfo
# 可以查询到CPU、GPU信息
~$ /opt/rocm/opencl/bin/clinfo
# 可以查询到GPU信息
若以上三条命令都有正常输出,便是配置成功了
(7)接下来,如果有利用GPU进行pytorch深度学习的需求,可以进行python程序验证,简单的指令例如:
>>>import torch
>>> torch.cuda.is_available()
返回True就说明成功啦
版权归原作者 WocheDonner 所有, 如有侵权,请联系我们删除。