安装 ROCm 4.0.1

Installing ROCm 4.0.1

在最新更新中,Pytorch 提供了对 ROCm 4.0.1 的支持。 我尝试使用官方 AMD site 上的指南安装最新版本的 ROCm,但这样做我只安装了 4.0.0 版。

据我了解这一步

wget -q -O - https://repo.radeon.com/rocm/rocm.gpg.key | sudo apt-key add -

echo 'deb [arch=amd64] https://repo.radeon.com/rocm/apt/debian/ xenial main' | sudo tee /etc/apt/sources.list.d/rocm.list

我应该更改存储库,但我不知道如何更改。

有人可以帮我确定我应该 运行 安装 4.0.1 的确切行数吗?

看来 https://repo.radeon.com/rocm/apt/debian/ 的 apt 存储库的维护者尚未在那里上传 rocm 4.0.1 的软件包。

如果您需要这个确切的版本,您可以使用 http://repo.radeon.com/rocm/apt/4.0.1/ 的存储库,而不是按照您链接到的指南中的建议。请注意,这会将您的 rocm 版本固定到此版本,而不会进行任何自动更新。