一、安裝pip和vim

sudo apt-get -y install python3-pip vim git

二、配置pip源

mkdir ~/.pip && vim ~/.pip/pip.conf

pip.conf

[global]
index-url = https://mirrors.cloud.tencent.com/pypi/simple
[install]
trusted-host = mirrors.cloud.tencent.com

三、安裝顯卡驅(qū)動(dòng)和pytorch?

# rocm5.4.2需要Ubuntu內(nèi)核5.15+
$ cd ~ && uname -srmv

• Ubuntu 22.04 + rocm 5.4.2

# 安裝顯卡驅(qū)動(dòng)
$ wget https://repo.radeon.com/amdgpu-install/5.4.2/ubuntu/jammy/amdgpu-install_5.4.50402-1_all.deb 
$ sudo apt-get install ./amdgpu-install_5.4.50402-1_all.deb
$ sudo apt-get update
$ amdgpu-install -y --usecase=graphics,rocm

# 設(shè)置運(yùn)行權(quán)限
$ ls -l /dev/dri/render*
$ sudo usermod -a -G render $LOGNAME
$ sudo usermod -a -G video $LOGNAME

# 安裝pytorch
$ pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/rocm5.4.2

# 重啟
$ reboot

• ?Ubuntu 20.04 + rocm 5.4.2

$ wget https://repo.radeon.com/amdgpu-install/5.4.2/ubuntu/focal/amdgpu-install_5.4.50402-1_all.deb
$ sudo apt-get install ./amdgpu-install_5.4.50402-1_all.deb
$ sudo apt-get update
$ amdgpu-install -y --usecase=graphics,rocm

# 設(shè)置運(yùn)行權(quán)限
$ ls -l /dev/dri/render*
$ sudo usermod -a -G render $LOGNAME
$ sudo usermod -a -G video $LOGNAME

$ pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/rocm5.4.2

# 重啟
$ reboot

• ?Ubuntu 20.04 + rocm 5.2

$ wget https://repo.radeon.com/amdgpu-install/22.40.3/ubuntu/focal/amdgpu-install_5.4.50403-1_all.deb
$ sudo apt-get install ./amdgpu-install_5.4.50403-1_all.deb
$ sudo apt-get update
$ amdgpu-install -y --usecase=graphics

$ sudo apt-get purge amdgpu-install

$ wget https://repo.radeon.com/amdgpu-install/22.20/ubuntu/focal/amdgpu-install_22.20.50200-1_all.deb 
$ sudo apt-get install ./amdgpu-install_22.20.50200-1_all.deb
$ sudo apt-get update
$ amdgpu-install -y --usecase=rocm

# 設(shè)置運(yùn)行權(quán)限
$ ls -l /dev/dri/render*
$ sudo usermod -a -G render $LOGNAME
$ sudo usermod -a -G video $LOGNAME

$ pip3 install torch==1.13.1+rocm5.2 torchvision==0.14.1+rocm5.2 torchaudio==0.13.1 --extra-index-url https://download.pytorch.org/whl/rocm5.2

# 重啟
$ reboot

四、檢測顯卡驅(qū)動(dòng)是否安裝成功

# 顯示GPU活動(dòng)信息
$ rocm-smi
# 檢查內(nèi)核模式驅(qū)動(dòng)程序的安裝
$ dkms status

# 如果兩個(gè)命令都列出了GPU，則認(rèn)為安裝成功
$ /opt/rocm-5.4.2/bin/rocminfo
# 或者
$ /opt/rocm-5.4.2/opencl/bin/clinfo

五、檢測pytorch是否安裝成功

當(dāng)顯示為True時(shí)，才能證明rocm安裝成功

$ vim ~/check.py
$ python3 ~/check.py

?check.py

import torch

print(torch.cuda.is_available())

六、錯(cuò)誤解決

• "hipErrorNoBinaryForGpu: Unable to find code object for all current devices!" 已放棄 (核心已轉(zhuǎn)儲)

check.py?

import torch
import os

os.environ["HSA_OVERRIDE_GFX_VERSION"] = "10.3.0"

print(torch.cuda.is_available())

?加入環(huán)境變量

# 對全局用戶生效
$ sudo vim /etc/profile
$ source /etc/profile

# 對當(dāng)前用戶生效
$ vim ~/.bash_profile
$ source ~/.bash_profile

?.bash_profile末尾加入

export HSA_OVERRIDE_GFX_VERSION=10.3.0

參考資料

pytorch官網(wǎng)
AMD顯卡驅(qū)動(dòng)下載

AMD顯卡驅(qū)動(dòng)安裝文檔

ROCm安裝文檔文章來源地址http://www.zghlxwxcb.cn/news/detail-444721.html

到了這里，關(guān)于Ubuntu22.04安裝AMD顯卡驅(qū)動(dòng)和pytorch的文章就介紹完了。如果您還想了解更多內(nèi)容，請?jiān)谟疑辖撬阉鱐OY模板網(wǎng)以前的文章或繼續(xù)瀏覽下面的相關(guān)文章，希望大家以后多多支持TOY模板網(wǎng)！