【安装学习】ubuntu环境配置AMD显卡驱动

ubuntu环境配置AMD显卡驱动

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

• 针对GEM12 MAX 板载的显卡680M，应当选择6.1版本的.deb文件

  https://repo.radeon.com/amdgpu-install/6.1/ubuntu/focal/

• 对应的pytorch安装命令，应当修改为：

  pip install torch==2.4.1 torchvision==0.19.1 torchaudio==2.4.1 --index-url https://download.pytorch.org/whl/rocm6.1

(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程序验证，简单的指令例如：

python3
>>> import torch
>>> torch.cuda.is_available()

返回True就说明成功啦