在上一篇文章中，簡單介紹了一下ROCm，這篇詳細說說如何在MD Radeon RX 7000/6000系列顯卡安裝ROCm 調(diào)用CUDA。

因為我的主機是AMD 6950XT，正好以我自己的主機為例做環(huán)境部署。

首先需要安裝雙系統(tǒng)，這里我以自己安裝的為例，為了方便推廣并寫入了關于7000系顯卡的部署教程：

安裝ROCm
RX6000系列及以下顯卡使用ROCm 5.4.2穩(wěn)定版本命令
【支持包括桌面級AMD Radeon RX6950XT、RX6900XT、RX6800XT、RX6800、RX6750XT、RX6750GRE、RX6700XT、RX6700GRE、RX6700、RX6650XT、RX6600XT、RX6600、RX6500XT、RX6400等
移動端AMD Radeon RX6850M XT、RX6800M、RX6700M、RX6650M XT、RX6800S、RX6700S、RX6650M、RX6600S、RX6600M、RX6500M、RX6300M等】
pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/rocm5.4.2

較新的RX7000系列顯卡需要安裝ROCm 5.6版本(加上–pre選項是查詢最新的預發(fā)布或開發(fā)版本)
【支持包括桌面級AMD Radeon RX7900 XTX、RX7900 XT、RX7800 XT、RX7700 XT、Radeon RX7600等
移動端RX 7600M XT等】
pip install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm5.6

2. 安裝完成在python環(huán)境里驗證cuda是否可用：

import torch
torch.cuda.is_available()

代碼執(zhí)行結(jié)果打印true說明是OK的

3.運行你的項目之前，有一項必要的環(huán)境變量配置：

3.1 首先打開Linux配置文件：

vim ~/.bashrc

3.2 在末尾加上一行：
若是RX6000系列及以下顯卡復制粘貼這個：
export HSA_OVERRIDE_GFX_VERSION=10.3.0

若是RX7000系列顯卡則復制粘貼這個：
export HSA_OVERRIDE_GFX_VERSION=11.0.0

3.3 保存配置并退出，使配置立即生效：
source ~/.bashrc