在 Windows 11 上运行 AMD ROCm™ 软件:综合指南

realtime news  Jul 24, 2024 14:25  UTC 06:25

1 Min Read

根据 AMD.com 的说法,最新的 AMD 软件 24.6.1 和 AMD ROCm™ 6.1.3 现在支持在 Windows 上使用 AMD Radeon™ RX 7000 系列显卡通过硬件加速运行 Linux 应用程序。这一发展为希望在 Windows 平台上利用强大机器学习和 AI 工具(如 Stable Diffusion)的用户提供了新的可能性。

在 Windows 11 上运行 AMD ROCm™ 的分步指南

以下是帮助您在 Windows 11 上运行 AMD ROCm™ 软件的详细指南:

步骤 1:安装 Windows Subsystem for Linux (WSL)

首先通过 Windows Powershell 安装 WSL。使用以下命令:

wsl --install

此命令将在您的 Windows 机器上设置运行 Linux 应用程序所需的环境。

步骤 2:安装 AMD ROCm™ 软件

接下来,继续安装 AMD ROCm™ 软件。按照以下命令完成安装:

sudo apt update
wget https://repo.radeon.com/amdgpu-install/6.1.3/ubuntu/jammy/amdgpu-install_6.1.60103-1_all.deb
sudo apt install ./amdgpu-install_6.1.60103-1_all.deb
amdgpu-install -y --usecase=wsl,rocm --no-dkms

这些步骤将确保 ROCm™ 软件正确安装并配置以便与 WSL 配合使用。

步骤 3:安装适用于 AMD ROCm™ 和 WSL 的 PyTorch

最后,安装适用于 AMD ROCm™ 软件的 Torch。使用以下命令:

sudo apt install python3-pip -y
pip3 install --upgrade pip wheel
wget https://repo.radeon.com/rocm/manylinux/rocm-rel-6.1.3/torch-2.1.2%2Brocm6.1.3-cp310-cp310-linux_x86_64.whl
wget https://repo.radeon.com/rocm/manylinux/rocm-rel-6.1.3/torchvision-0.16.1%2Brocm6.1.3-cp310-cp310-linux_x86_64.whl
wget https://repo.radeon.com/rocm/manylinux/rocm-rel-6.1.3/pytorch_triton_rocm-2.1.0%2Brocm6.1.3.4d510c3a44-cp310-cp310-linux_x86_64.whl
pip3 uninstall torch torchvision pytorch-triton-rocm numpy torchaudio
pip3 install torch-2.1.2+rocm6.1.3-cp310-cp310-linux_x86_64.whl torchvision-0.16.1+rocm6.1.3-cp310-cp310-linux_x86_64.whl pytorch_triton_rocm-2.1.0+rocm6.1.3.4d510c3a44-cp310-cp310-linux_x86_64.whl numpy==1.26.4
pip3 install torchaudio==2.1.2

完成这些安装后,执行以下命令以完成设置:

location=`pip show torch | grep Location | awk -F ": " '{print $2}'`
cd ${location}/torch/lib/
rm libhsa-runtime64.so*
cp /opt/rocm/lib/libhsa-runtime64.so.1.2 libhsa-runtime64.so

这些命令确保 ROCm™ 库链接正确并准备就绪。

运行应用程序

设置完成后,您现在可以安装并运行支持 AMD ROCm™ 软件的应用程序,而无需离开 Windows 环境。例如,像 ComfyUI 这样的应用程序现在可以在 AMD Radeon™ RX 7900 XTX 上无缝运行。

注意:在安装新应用程序时,请检查 requirements.txt 文件并注释掉 torch 行,以避免覆盖您的 AMD ROCm™ 软件版本。否则,可能需要重新安装这些版本。

这一新功能显著增强了 Windows 11 的多功能性和实力,适合那些利用 AMD 硬件进行高级计算任务的用户。



Read More