欢迎光临散文网 会员登陆 & 注册

Ubuntu20.04.5上意义不明的AMD显卡Rocm教程(补)

2022-10-27 09:57 作者:游手好闲的黑夜sama  | 我要投稿

实验环境:

系统:Ubuntu20.04.5 LTS 桌面版

显卡:AMD RX6800XT 

 

部署成功后也可以在评论区发出你的显卡型号和系统。经多数人测试RX6000系列、RX6000M系列、RX6000S系列、RX Vega系列、5000系列均可使用!


此教程是看了国内外各种教程缝合后以及经过评论区小伙伴的各种反馈后反复修改后的成果,在这里感谢!!!
非虚拟化环境!!!分个150g空间安装ubuntu,并且最好单分区,可以与win共存!想用哪个系统就重启切换!
教程比较长,并且无图,可能需要会一点linux的人才能看懂,并且为了兼容多数设备已修改多次。但是按照顺序下来绝对不会错(如果硬件和系统不一致就无法保证),所以请耐心慢慢看!


 

一、更新apt并安装一些需要的程序


二、安装amdgpu驱动以及rocm


漫长的等待...



此时可以通过下面命令确认安装(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

如果出现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

进行验证

python3
import torch
torch.cuda.is_available()

输出结果为 True 即可以调用gpu
然后按CTRL+D退出python

↑↑↑↑↑↑

###可选忽略步骤###

 

现在可以下载安装本体了!!

这时候就去把需要的模型以及必要文件放进各个文件夹具体参考:


直到出现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

 

 

 


Ubuntu20.04.5上意义不明的AMD显卡Rocm教程(补)的评论 (共 条)

分享到微博请遵守国家法律