如何修复暂停后 CUDA 中断的问题？

Question

正如标题所说，每次我将笔记本电脑从挂起状态唤醒时，CUDA 都会崩溃，根本无法工作。我必须重新启动系统才能将其恢复。因此，要在暂停后恢复它，我必须运行“sudo rmmod nvidia_uvm”，然后运行“sudo modprobe nvidia_uvm”，当然这可以工作，但正如你想象的那样，它很快就会过时。那么如何永久恢复呢？如果有什么区别的话，我有一台 PRIME 笔记本电脑，我的 Nvidia GPU 始终选择 (940MX) 和 440 驱动程序。

Answer 1

看起来该nvidia_uvm模块在挂起时无法正常工作。如果重新加载模块，CUDA可以再次使用：

sudo rmmod nvidia_uvm
sudo modprobe nvidia_uvm

不幸的是，当 CUDA 应用程序处于活动状态时，这不起作用。

我已提交错误报告：https://bugs.launchpad.net/ubuntu/+source/nvidia-graphics-drivers-460/+bug/1921477

Answer 2

我在ubuntu上也遇到同样的问题。最近转向Archlinux。按照wiki 上的提示进行操作，它已在我的 gtx1650 笔记本电脑上解决了此问题。

这样人们就可以激活一种新的电源管理工具。请参阅nvidia 文档。

我的步骤：

1. 创建/etc/modprobe.d/nvidia-power-management.conf与options nvidia NVreg_PreserveVideoMemoryAllocations=1 NVreg_TemporaryFilePath=/tmp.
2. 更新initramfs。在 ubuntu 上是通过update-initramfs -u
3. systemctl enable nvidia-suspend.service。我认为你必须首先在 Ubuntu 上手动安装 systemd 服务文件。看这个帖子
4. 重启