本人買的AMD顯卡,最近要在ubuntu環(huán)境做python深度學(xué)習(xí),所以必須折騰一番,結(jié)果發(fā)現(xiàn)還不復(fù)雜。本人的系統(tǒng)是Ubuntu 20.04.6,顯卡是RX6600,親測有效。
(0) 如果已經(jīng)配置了顯卡驅(qū)動但是沒有成功,或配置成功想更換版本,則先卸載原有的驅(qū)動,重新開始:
~$ sudo amdgpu-install --uninstall
(1) 在AMD官網(wǎng)下載需要的驅(qū)動包:
驅(qū)動包鏈接
得到amdgpu-install_xxx_all.deb(xxx是版本號)
(2) cd到驅(qū)動包路徑下,安裝驅(qū)動包:
~$ sudo apt install ./amdgpu-install_xxx_all.deb
(3) 更新軟件:
~$ sudo apt update
~$ sudo apt upgrade -y
(4) 安裝驅(qū)動:
~$ sudo amdgpu-install --no-dkms
~$ sudo apt install rocm-dev
~$ sudo reboot
# 等待重啟
(5)配置環(huán)境:
~$ ls -l /dev/dri/render*
~$ sudo usermod -a -G render $LOGNAME
~$ sudo usermod -a -G video $LOGNAME
# $LOGNAME不需要修改
~$ sudo reboot
# 等待重啟
(6) 理論上如果一路上沒有報錯,便配置成功了,可以輸入下列命令驗證
~$ rocm-smi
# 監(jiān)控GPU性能
~$ /opt/rocm/bin/rocminfo
# 可以查詢到CPU、GPU信息
~$ /opt/rocm/opencl/bin/clinfo
# 可以查詢到GPU信息
若以上三條命令都有正常輸出,便是配置成功了文章來源:http://www.zghlxwxcb.cn/news/detail-753038.html
(7)接下來,如果有利用GPU進行pytorch深度學(xué)習(xí)的需求,可以進行python程序驗證,簡單的指令例如:
>>> import torch
>>> torch.cuda.is_available()
返回True就說明成功啦文章來源地址http://www.zghlxwxcb.cn/news/detail-753038.html
到了這里,關(guān)于ubuntu環(huán)境配置AMD顯卡驅(qū)動的文章就介紹完了。如果您還想了解更多內(nèi)容,請在右上角搜索TOY模板網(wǎng)以前的文章或繼續(xù)瀏覽下面的相關(guān)文章,希望大家以后多多支持TOY模板網(wǎng)!