ubuntu环境配置AMD显卡驱动

本人买的AMD显卡，最近要在ubuntu环境做python深度学习，所以必须折腾一番，结果发现还不复杂。本人的系统是Ubuntu 20.04.6，显卡是RX6600，亲测有效。

(0) 如果已经配置了显卡驱动但是没有成功，或配置成功想更换版本，则先卸载原有的驱动，重新开始：

~$ sudo amdgpu-install --uninstall

(1) 在AMD官网下载需要的驱动包：

驱动包链接
得到amdgpu-install_xxx_all.deb（xxx是版本号）

(2) cd到驱动包路径下，安装驱动包：

~$ sudo apt install ./amdgpu-install_xxx_all.deb

(3) 更新软件：

~$ sudo apt update~$ sudo apt upgrade -y

(4) 安装驱动：

~$ sudo amdgpu-install --no-dkms~$ sudo apt install rocm-dev~$ sudo reboot# 等待重启

(5)配置环境：

~$ ls -l /dev/dri/render*~$ sudo usermod -a -G render $LOGNAME~$ sudo usermod -a -G video $LOGNAME# $LOGNAME不需要修改~$ sudo reboot# 等待重启

(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就说明成功啦

本文链接：https://www.kjpai.cn/news/2024-04-18/160179.html，文章来源：网络cs，作者：言安琪，版权归作者所有，如需转载请注明来源和作者，否则将追究法律责任！