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

评论 9
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

WocheDonner

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值