我用的操作系統(tǒng)為ubuntu20.04,其他系統(tǒng)應(yīng)該類似，只是命令稍有不同。
安裝AMD GPU驅(qū)動(dòng)
AMD驅(qū)動(dòng)下載地址:https://www.amd.com/en/support/kb/release-notes/rn-amdgpu-unified-linux-22-20
進(jìn)去之后找到你想安裝的版本,這里是閉源驅(qū)動(dòng)。
下載好后運(yùn)行sudo apt install amdgpu-install_5.4.50403-1_all.deb
這個(gè)名字根據(jù)自己下載的文件進(jìn)行更改，直接Tab就行。
如果報(bào)依賴錯(cuò)誤可以嘗試sudo apt-get -f install命令來(lái)修復(fù)依賴錯(cuò)誤。
之后更新依賴sudo apt update.
安裝ROCM
安裝ROCM，這里要--no-dkms不安裝內(nèi)核，看資料好像現(xiàn)在的ubuntu自己帶內(nèi)核，所以加上好像會(huì)有問(wèn)題。有知道的可以評(píng)論區(qū)討論。
sudo amdgpu-install --usecase=rocm,hip,mllib --no-dkms
到這里驅(qū)動(dòng)和ROCM就安裝完了。
安裝Pytorch
可以在pytorch官網(wǎng)找到安裝命令
https://pytorch.org/
?
例如我用pip安裝
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/rocm5.4.2
測(cè)試安裝
打開(kāi)python3
import torch
torch.cuda.is_available()
如果返回True
恭喜你，安裝成功。
安裝完之后可能還需要進(jìn)行一些操作。
#添加環(huán)境
sudo usermod -a -G video $LOGNAME
sudo usermod -a -G render $LOGNAME
#在文件執(zhí)行
vim ~/.bashrc #添加下面
export HSA_OVERRIDE_GFX_VERSION=10.3.0 #我這里的顯卡是amd的RX6700xt
export LD_LIBRARY_PATH=/opt/rocm/lib
source ~/.bashrc#通重新執(zhí)行剛修改的初始化文件
# 如果運(yùn)行import之后報(bào)錯(cuò)找不到這些文件，復(fù)制一下鏈接文件
cd $LD_LIBRARY_PATH
cp librocblas.so.0 librocblas.so
cp libMIOpen.so.1 libMIOpen.so
cp librocrand.so.1 librocrand.so
其它問(wèn)題：
Process finished with exit code 139 (interrupted by signal 11: SIGSEGV)
我還遇到了這個(gè)報(bào)錯(cuò)，原因是我沒(méi)有添加
export HSA_OVERRIDE_GFX_VERSION=10.3.0 #我這里的顯卡是amd的RX6700xt
export LD_LIBRARY_PATH=/opt/rocm/lib
到對(duì)應(yīng)的用戶下，只加了root，所以只有root能用。
有熱心網(wǎng)友指出：文章來(lái)源：http://www.zghlxwxcb.cn/news/detail-597714.html
dkms可以安裝。跟系統(tǒng)內(nèi)核不沖突，只有在裝多版本ROCm時(shí)不能裝dkms。
export HSA_OVERRIDE_GFX_VERSION=10.3.0一般不用。我的同學(xué)時(shí)7900xtx被這個(gè)坑了一下，不用export就行。
也不用export LD_LIBRARY_PATH=/opt/rocm/lib ，hipconfig命令執(zhí)行應(yīng)該能獲得準(zhǔn)確的環(huán)境變量。rocmdev和mlsdk也可以安裝一下。文章來(lái)源地址http://www.zghlxwxcb.cn/news/detail-597714.html
到了這里，關(guān)于AMD GPU驅(qū)動(dòng),ROCM,Pytorch安裝教程(A卡6700xt)的文章就介紹完了。如果您還想了解更多內(nèi)容，請(qǐng)?jiān)谟疑辖撬阉鱐OY模板網(wǎng)以前的文章或繼續(xù)瀏覽下面的相關(guān)文章，希望大家以后多多支持TOY模板網(wǎng)！
