Ubuntu20.04.5上意義不明的AMD顯卡Rocm教程(補)
實驗環(huán)境:
系統(tǒng):Ubuntu20.04.5?LTS 桌面版
顯卡:AMD RX6800XT?
?
部署成功后也可以在評論區(qū)發(fā)出你的顯卡型號和系統(tǒng)。經(jīng)多數(shù)人測試RX6000系列、RX6000M系列、RX6000S系列、RX Vega系列、5000系列均可使用!
此教程是看了國內(nèi)外各種教程縫合后以及經(jīng)過評論區(qū)小伙伴的各種反饋后反復(fù)修改后的成果,在這里感謝?。?!
非虛擬化環(huán)境!??!分個150g空間安裝ubuntu,并且最好單分區(qū),可以與win共存!想用哪個系統(tǒng)就重啟切換!
教程比較長,并且無圖,可能需要會一點linux的人才能看懂,并且為了兼容多數(shù)設(shè)備已修改多次。但是按照順序下來絕對不會錯(如果硬件和系統(tǒng)不一致就無法保證),所以請耐心慢慢看!

?
一、更新apt并安裝一些需要的程序
二、安裝amdgpu驅(qū)動以及rocm
漫長的等待...
此時可以通過下面命令確認(rèn)安裝(6800xt會顯示gfx1030,其他的我就不知道了)
###可選忽略步驟###
↓↓↓↓↓↓
wget https://download.pytorch.org/whl/rocm5.1.1/torch-1.12.1%2Brocm5.1.1-cp38-cp38-linux_x86_64.whl
sudo pip install torch-1.12.1+rocm5.1.1-cp38-cp38-linux_x86_64.whl
如果出現(xiàn)read timed out的話,上面的命令在來一遍
sudo pip install torch torchvision --extra-index-url https://download.pytorch.org/whl/rocm5.1.1
?
export HSA_OVERRIDE_GFX_VERSION=10.3.0
進(jìn)行驗證
python3
import torch
torch.cuda.is_available()
輸出結(jié)果為 True 即可以調(diào)用gpu
然后按CTRL+D退出python
↑↑↑↑↑↑
###可選忽略步驟###
?
現(xiàn)在可以下載安裝本體了!!
這時候就去把需要的模型以及必要文件放進(jìn)各個文件夾具體參考:

直到出現(xiàn)127.0.0.1:7860時,說明完成
然后就可以打開瀏覽器訪問啦!
下次啟動只要輸入?
如果想在A卡上煉丹的話(未測試)請看:


以下為此方案參考資料:
https://rocmdocs.amd.com/en/latest/Installation_Guide/Installation_new.html#rocm-installation-guide-v5-0
https://docs.amd.com/bundle/ROCm_Installation_Guidev5.0/page/How_To_Install_ROCm.html
https://github.com/ROCmSoftwarePlatform/MIOpen
https://www.cnblogs.com/lllzhuang/articles/16083003.html
https://github.com/RadeonOpenCompute/ROCm/issues/1756
https://github.com/AUTOMATIC1111/stable-diffusion-webui/wiki/Install-and-Run-on-AMD-GPUs
?
?
?