Ubuntu下AMD显卡安装amd显卡驱动以及Rocm的教程

March 28, 2024 by Clyde

AMD显卡Rocm教程

1

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

sudo apt-get update
sudo apt-get upgrade
sudo apt-get install wget gnupg2 git python3 python3-venv python3-pip

二` 安装amdgpu驱动以及rocm
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 amdgpu-install --usecase=hiplibsdk,rocm --no-dkms

三`添加系统到路径
echo 'export PATH=$PATH:/opt/rocm/bin:/opt/rocm/profiler/bin:/opt/rocm/opencl/bin' | sudo tee -a /etc/profile.d/rocm.sh

此时可以通过下面命令确认安装(6800xt会显示gfx1030,其他的我就不知道了)

rocm-smi
sudo /opt/rocm/bin/rocminfo
sudo /opt/rocm/opencl/bin/clinfo

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

git clone https://github.com/AUTOMATIC1111/stable-diffusion-webui
cd stable-diffusion-webui
python3 -m venv venv
source venv/bin/activate
export HSA_OVERRIDE_GFX_VERSION=10.3.0
python3 -m pip install --upgrade pip wheel

TORCH_COMMAND='pip install torch torchvision --extra-index-url https://download.pytorch.org/whl/rocm5.1.1' python launch.py --precision full --no-half

直到出现127.0.0.1:7860时,说明完成

然后就可以打开浏览器访问啦!

下次启动只要输入 

source venv/bin/activate
export HSA_OVERRIDE_GFX_VERSION=10.3.0(非原生gfx1030输入命令)
python launch.py --precision full --no-half

如需评论,请填写表单。

提交前先勾选

© 2024 | The Asuna Blog From | X-ways theme