在Windows 11上安装ROCm 10

ROCm版本号一下子跳到10了（目前最新版10.1，也是本文基于的版本），据说性能也有大幅提升，而安装方法也发生了变化。这一次官方终于给出了pip包仓库，不用再下载文件来安装了。英文水平可以或需要其他平台安装方法的，可以访问https://rocm.docs.amd.com/en/latest/install/rocm.html，下面仅说明Windows 11下的安装方法。

兼容性及驱动

ROCm 10支持的显卡请参考这里，目前看来，桌面显卡的RX 9000系列、RX 7000系列，以及APU集成的8000S系列、800M系列、700M系列都是支持的，还有专业卡、计算卡就不细说了。

驱动需要安装最新的AMD Software: Adrenalin Edition for ROCm 10.1，不确定后续会不会加入到常规驱动里。

如果你只是想玩玩本地大模型或图像生成，那么可以安装LM Studio或Comfy Desktop，两者都对AMD显卡支持不错，就不用继续往下了。

另外官方支持的Windows版本是Windows 11 25H2，其他版本兼容性不确定（虽然理论上来说问题不大）。

安装Python

目前只支持Python，版本为3.11、3.12、3.13或3.14，终于不再只是3.12了。关于Python虚拟环境管理工具，可以参看Windows安装Miniconda教程，后续命令都基于Miniconda。

新建虚拟环境：

conda create -n pytorch python=3.14

create是新建虚拟环境；-n是设置虚拟环境名称；pytorch为虚拟环境名称，请自行命名；后面安装的Python版本根据需求自选。

然后激活虚拟环境：

conda activate pytorch

安装ROCm SDK

python -m pip install --index-url https://stable.repo.amd.com/rocm/whl-next/ "rocm[libraries,device-all]==10.1.0"

这是官方给出的命令，理解含义的可自行尝试，我就不修改了。

安装完后可以验证安装：

hipinfo

这是查看显卡信息和驱动是否安装到位。

pip freeze
where rocm-sdk

这是查看SDK是否正确安装。

rocm-sdk test

这是测试SDK。

安装PyTorch

官方也给出了PyTorch、JAX和TenserFlow的安装指南，这里就安装更常用的PyTorch。

在前面已装好ROCm SDK的虚拟环境下安装：

python -m pip install --index-url https://stable.repo.amd.com/rocm/whl-next/ "torch[device-all]==2.14.0+rocm10.1.0" "torchvision[device-all]==0.29.0a0+rocm10.1.0" "torchaudio==2.11.0.3+rocm10.1.0"

可见同样是在官方的pip包仓库里安装。这每个版本换一种安装方法，AMD你有点延续性好不好！

卸载

如果激活了虚拟环境，则先取消激活：

conda deactivate

然后删除虚拟环境即可：

conda env remove -n pytorch

卸载驱动则别忘了把相关的HIP SDK相关的包卸载了，然后装回普通驱动即可。