5

我想在 Ubuntu 上设置 AMD Radeon 进行深度学习。我工作的主要库是 keras 和 pytorch。我在这里严格遵循 ROCm 安装指南,但在第三步使用命令失败sudo apt install rocm-dkms。错误信息如下所示。

Setting up dkms (2.8.1-5ubuntu1) ...
Setting up hip-rocclr (4.0.20496.5685.40000-23) ...
Setting up rock-dkms (1:4.0-23) ...
Loading new amdgpu-4.0-23 DKMS files...
Building for 5.8.0-41-generic
Building for architecture x86_64
Building initial module for 5.8.0-41-generic
Error! Bad return status for module build on kernel: 5.8.0-41-generic (x86_64)
Consult /var/lib/dkms/amdgpu/4.0-23/build/make.log for more information.
dpkg: error processing package rock-dkms (--configure):
 installed rock-dkms package post-installation script subprocess returned error 
exit status 10
Setting up g++-9 (9.3.0-17ubuntu1~20.04) ...
Setting up g++ (4:9.3.0-1ubuntu2) ...
update-alternatives: using /usr/bin/g++ to provide /usr/bin/c++ (c++) in auto mo
de
Setting up build-essential (12.8ubuntu1.1) ...
dpkg: dependency problems prevent configuration of rocm-dkms:
 rocm-dkms depends on rock-dkms; however:
  Package rock-dkms is not configured yet.

dpkg: error processing package rocm-dkms (--configure):
 dependency problems - leaving unconfigured
Setting up gcc-multilib (4:9.3.0-1ubuntu2) ...
No apport report written because the error message indicates its a followup erro
r from a previous failure.
                          Setting up g++-9-multilib (9.3.0-17ubuntu1~20.04) ...
Setting up g++-multilib (4:9.3.0-1ubuntu2) ...
Processing triggers for sgml-base (1.29.1) ...
Setting up x11proto-dev (2019.2-1ubuntu1) ...
Setting up libxau-dev:amd64 (1:1.0.9-0ubuntu1) ...
Processing triggers for libc-bin (2.31-0ubuntu9.2) ...
Processing triggers for man-db (2.9.1-1) ...
Setting up libxdmcp-dev:amd64 (1:1.1.3-0ubuntu1) ...
Setting up x11proto-core-dev (2019.2-1ubuntu1) ...
Setting up libxcb1-dev:amd64 (1.14-2) ...
Setting up libx11-dev:amd64 (2:1.6.9-2ubuntu1.1) ...
Setting up libglx-dev:amd64 (1.3.2-1~ubuntu0.20.04.1) ...
Setting up libgl-dev:amd64 (1.3.2-1~ubuntu0.20.04.1) ...
Setting up mesa-common-dev:amd64 (20.2.6-0ubuntu0.20.04.1) ...
Setting up rocm-opencl-dev (3.6Beta-17-g875c1f8-rocm-rel-4.0-23) ...
Settin XT g up rocm-clang-ocl (0.5.0.64-rocm-rel-4.0-23-50fb51a) ...
Setting up rocm-utils (4.0.0.40000-23) ...
Setting up rocm-dev (4.0.0.40000-23) ...
Processing triggers for libc-bin (2.31-0ubuntu9.2) ...
Errors were encountered while processing:
 rock-dkms
 rocm-dkms
E: Sub-process /usr/bin/dpkg returned an error code (1)

我的内核版本是5.8.0-41-generic. 我的显卡是技嘉Radeon RX6900 XT。我的 CPU 是 AMD Ryzen 9 3900 XT。我尝试了以前帖子中建议的几种解决方案,但并没有解决我的问题。我可以有你的建议来解决这个问题。

4

2 回答 2

8

我也遇到了同样的问题。我发现修复它的唯一方法是回滚到 5.6.0-1042-oem 内核。AMD 驱动程序似乎不支持除此之外的任何内核。

编辑:这也是让 amdgpupro 驱动程序安装没有问题的一种方法。

警告:我是在事后写下所有这些,我可能在此过程中错过了一步或其他东西。请非常小心,尤其是在尝试删除内核以及在引导目录中工作时。如果您对破坏系统的想法感到不舒服,您可以随时设置 grub 的默认选择,这比删除 initramfs 安全得多。

这是我如何让 RocM 工作的

sudo apt install linux-image-5.6.0-1042-oem linux-headers-5.6.0-1042-oem && reboot

确保通过访问 grub 中的 Ubuntu 高级选项启动到 5.6 内核。

sudo apt remove linux-image-5.8.0-41-generic linux-headers-5.8.0-41-generic && sudo apt autoremove && reboot

同样,您必须通过高级选项重新启动到 5.6。(在 BIOS 完成加载后按住 shift 键以获取 Ubuntu 高级选项菜单。)在您返回后,最好将您的标题和图像设置为保留,因为内核更新很可能会破坏 RocM。

sudo apt-mark hold linux-image-generic linux-headers-generic

现在我们将尝试清除 5.8 内核。首先清除临时文件。

sudo rm -rv ${TMPDIR:-/var/tmp}/mkinitramfs-*

现在列出所有已安装的内核。

dpkg -l | tail -n +6 | grep -E 'linux-image-[0-9]+'

并尝试删除 5.8 内核。对我们安装的 5.6 以上的任何内核执行此操作。

sudo update-initramfs -d -k 5.8.0-41-generic

现在 initramfs、Systemmap 和 config 仍然存在于引导目录中,因此我们需要清除它们以使 grub 再次正常工作。

cd /boot/
sudo rm vmlinuz-5.8.0-41-generic System.map-5.8.0-41-generic config-5.8.0-41-generic

现在你应该终于准备好更新 grub了

sudo update-grub && reboot

现在,当您重新加载时,您应该能够安装 RocM

sudo apt install rocm-dkms
于 2021-02-03T18:52:35.720 回答
2

根据此链接中的官方说明,AMD ROCm 平台旨在支持 Ubuntu 20.04.1(5.4 和 5.6-oem)和 18.04.5(内核 5.4)。

所以不支持内核版本 5.8。但是,降级是一种选择,但不要急于降级,您可以简单地启动到旧版本的内核。

尝试以下步骤:

  1. 重启你的电脑,
  2. 等待 grub 菜单打开(如何打开 grub 菜单:链接)。
  3. 选择advanced options for ubuntu
  4. 从显示的列表中选择一个备用内核。
于 2021-04-19T07:16:00.100 回答