nvidia-smi --gpu-reset命令有什么作用？是否仅用于释放GPU显存？

基础作用 #

这条命令是对索引为0的NVIDIA GPU执行硬件级重置操作，其中各部分含义如下：

• sudo：使用root权限执行操作，GPU重置需要管理员权限才能触发
• nvidia-smi：NVIDIA官方提供的GPU系统管理工具
• --gpu-reset：核心参数，触发GPU重置动作
• -i 0：指定操作目标为设备编号为0的GPU，你可以修改后面的数字操作其他GPU

能否仅释放显存？ #

该命令的作用远不止释放GPU显存，不推荐专门用它来清理显存，具体差异和适用场景如下：

• 它会清空目标GPU的所有运行状态：除了显存内的所有数据外，还会重置GPU的硬件寄存器、驱动层的异常状态、卡死的计算任务，相当于给GPU做一次热重启
• 它的设计用途是处理GPU异常场景：比如GPU进程卡死杀不掉、驱动报错不响应、显存被异常占用无法通过结束进程释放的问题
• 普通的显存释放需求，直接结束占用显存的对应进程即可完成，不需要执行风险更高的GPU重置操作

注意事项 #

• 执行命令前必须终止所有占用目标GPU的进程，否则可能引发驱动崩溃、系统卡顿甚至死机
• 已经被直通给虚拟机的GPU无法通过该命令重置
• 部分服务器硬件配置下，执行该命令后需要重启整机才能让GPU恢复正常工作

内容的提问来源于stack exchange，提问作者user14889957