0


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到驱动包路径下,安装驱动包:

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

标签: ubuntu linux python

本文转载自: https://blog.csdn.net/StruggleRay/article/details/133201391
版权归原作者 WocheDonner 所有, 如有侵权,请联系我们删除。

“ubuntu环境配置AMD显卡驱动”的评论:

还没有评论