达到 GPU 内存限制时重新启动 systemd 服务

达到 GPU 内存限制时重新启动 systemd 服务

我在 Ubuntu 22.04 LTS 上使用以下 systemd 服务。该服务保存在/lib/systemd/system/myService.service

[Unit]
Description=myService
RequiresMountsFor=/mnt/Data

[Service]
ExecStart=/usr/local/bin/project/main
Restart=always
RestartSec=5
StandardOutput=append:/usr/local/bin/project/stdout.log
StandardError=append:/usr/local/bin/project/stderr.log

[Install]
WantedBy=multi-user.target

我的服务正在使用 GPU。我想在达到一定数量的 GPU 内存使用量 (vram) 时自动重新启动服务。我可以使用MemoryHighandMemoryMax但我认为这仅适用于 CPU 内存。

当GPU内存超过限制一定时间(例如1分钟)时,是否可以自动重新启动服务?不应检测峰值,但仅当内存在较长时间内过高时才应检测到。

您还看到该服务还有其他改进吗?我希望该服务始终运行,即始终在启动时和崩溃时重新启动。我认为我已经通过服务文件实现了这一点。

相关内容