ubuntu20.04中配置AMD显卡（6600）pytorch

第一步下载amd安装工具

进入网站下载5.4.2的amd_gpu安装工具，5.4.2版本是需要与后面的pytorch版本对应。

Index of /amdgpu-install/5.4.2/ubuntu/focal/

下载完成，安装包移动到主目录下，运行命令安装

sudo apt-get install ./amdgpu-install_5.4.50402-1_all.deb

第二步安装rocm等待安装成功

sudo amdgpu-install --usecase=rocm,hip,mllib,dkms 

可以查看是否安装成功，没有报错就成功了：

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

第三步安装pytorch

pytorch官网PyTorch选择自己设备的情况，运行命令行，这里面选择ROCM5.4.2，注意5.4.2需要与amdgpu版本相同。

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

没有pip3的先安装pip3

第四步配置系统环境

#添加环境
 

sudo usermod -a -G video $LOGNAME
sudo usermod -a -G render $LOGNAME 

#在文件执行
在主目录下ctrl+h后，打开.bashrc文件
 

export HSA_OVERRIDE_GFX_VERSION=10.3.0
export LD_LIBRARY_PATH=/opt/rocm/lib
source ~/.bashrc#通重新执行刚修改的初始化文件 

然后重启电脑。

第五部检验，终端输入下面代码，并返回true成功

python3
import torch
torch.cuda.is_available()

参考：

AMD GPU驱动,ROCM,Pytorch安装教程(A卡6700xt) - 哔哩哔哩

AMD显卡使用Pytorch深度学习配置(RX6600 ubuntu20.04为例)-2022.9.20 - 知乎AMD显卡安装Pytorch