GPU 状态全解析从查看命令到显存泄漏排查与修复在运行强化学习训练时你是否遇到过CUDA out of memory错误明明 GPU 显存足够却在一段时间后崩溃本文将带你从基础命令开始深入分析 GPU 状态诊断显存泄漏并给出针对 RTX 3050 等 8GB 显存显卡的训练优化建议。一、查看 GPU 版本与状态的常用命令1. 查看驱动与 CUDA 版本最常用nvidia-smi输出示例Driver Version: 535.230.02 CUDA Version: 12.22. 只查看驱动版本nvidia-smi --query-gpudriver_version--formatcsv,noheader3. 查看 GPU 型号及详细信息nvidia-smi-L# 列出所有 GPUnvidia-smi-q|grepProduct# 仅显示型号4. 查看 PyTorch 使用的 CUDA 版本python-cimport torch; print(PyTorch CUDA版本:, torch.version.cuda)python-cimport torch; print(CUDA是否可用:, torch.cuda.is_available())python-cimport torch; print(GPU数量:, torch.cuda.device_count())5. 查看显存使用情况简洁nvidia-smi --query-gpumemory.total,memory.used,memory.free--formatcsv6. 查看 CUDA 工具包版本nvcc--version# 或cat/usr/local/cuda/version.txt7. 实时监控 GPU 状态watch-n1nvidia-smi二、实战分析当前 GPU 状态解读假设执行nvidia-smi后得到如下输出--------------------------------------------------------------------------------------- | NVIDIA-SMI 535.230.02 Driver Version: 535.230.02 CUDA Version: 12.2 | |------------------------------------------------------------------------------------- | GPU Name Persistence-M | Bus-Id Disp.A | Volatile Uncorr. ECC | | Fan Temp Perf Pwr:Usage/Cap | Memory-Usage | GPU-Util Compute M. | || | 0 NVIDIA GeForce RTX 3050 Off | 00000000:01:00.0 On | N/A | | 0% 50C P5 15W / 130W | 385MiB / 8192MiB | 28% Default | ------------------------------------------------------------------------------------- | Processes: | | GPU GI CI PID Type Process name GPU Memory | | ID ID Usage | || | 0 N/A N/A 1311 G /usr/lib/xorg/Xorg 35MiB | | 0 N/A N/A 1650 G /usr/lib/xorg/Xorg 169MiB | | 0 N/A N/A 1802 G /usr/bin/gnome-shell 38MiB | | 0 N/A N/A 29696 G ...cess-track-uuid... 57MiB | | 0 N/A N/A 31374 G /usr/share/code/code 49MiB | | 0 N/A N/A 35854 G ...WebKitWebProcess 15MiB | ---------------------------------------------------------------------------------------关键信息解读项目数值说明GPU 型号NVIDIA GeForce RTX 3050入门级光追显卡驱动版本535.230.02较新稳定版CUDA 版本12.2支持最新 PyTorch总显存8192 MiB (8 GiB)对强化学习训练偏小但可优化使用当前显存占用385 MiB仅系统图形界面和 VS Code 等进程可用显存~7807 MiB大量空闲GPU 温度50°C正常功耗15W / 130W低功耗状态P5未满载GPU 利用率28%当前无计算密集型任务✅结论此时 GPU 状态健康显存充足没有训练进程在运行。三、典型案例显存泄漏导致的CUDA out of memory错误日志回顾torch.OutOfMemoryError: CUDA out of memory. Tried to allocate 2.00 MiB. GPU 0 has a total capacity of 7.78 GiB of which 53.38 MiB is free. Including non-PyTorch memory, this process has 7.10 GiB memory in use. Of the allocated memory 6.99 GiB is allocated by PyTorch...崩溃前行为成功完成了约 20 个 episode1710 → 1720每个 episode 后打印Goal!!并 reset突然在actor_a_loss.backward()时显存耗尽根本原因显存泄漏Memory Leak不是过拟合而是训练代码中存在以下问题之一计算图未释放backward()后保留了中间变量经验回放缓冲区存储 GPU tensor每次采样都保留在显存中未及时清空 GPU 缓存长期累积导致碎片化网络保存时复制了计算图四、显存泄漏的修复方案1. 确保 backward 后释放计算图# 错误写法actor_loss.backward(retain_graphTrue)# 保留计算图会泄漏# 正确写法actor_loss.backward()# 默认 retain_graphFalse2. 经验回放缓冲区中存储 CPU tensor# 错误存储 GPU tensorself.memory.append((state,action,reward,next_state,done))# 正确转移到 CPUself.memory.append((state.cpu(),action.cpu(),reward,next_state.cpu(),done))# 使用时再转回 GPUstatestorch.stack([exp[0]forexpinbatch]).to(device)3. 手动清空 GPU 缓存# 在每个 episode 结束或每 N 步后调用torch.cuda.empty_cache()4. 使用with torch.no_grad()包装无梯度的计算withtorch.no_grad():target_actionsself.target_actor(next_states)target_qself.target_critic(next_states,target_actions)5. 减小 batch_size 和网络宽度batch_size64# 而不是 256hidden_dim256# 而不是 5126. 设置环境变量缓解显存碎片exportPYTORCH_CUDA_ALLOC_CONFexpandable_segments:True7. 完整 learn() 函数模板deflearn(self):iflen(self.memory)self.batch_size:returntorch.cuda.empty_cache()# 开始前清理# 采样并转移到 GPUstates,actions,rewards,next_states,donesself.sample_batch()statesstates.to(self.device)# ...# 计算 actor lossactor_lossself.compute_actor_loss(states)self.actor_optimizer.zero_grad()actor_loss.backward()self.actor_optimizer.step()# 计算 critic losscritic_lossself.compute_critic_loss(states,actions,rewards,next_states,dones)self.critic_optimizer.zero_grad()critic_loss.backward()self.critic_optimizer.step()torch.cuda.empty_cache()# 结束后清理五、针对 RTX 3050 (8GB 显存) 的训练建议超参数推荐值原因batch_size64 ~ 128太大容易爆显存hidden_size256网络宽度适中replay buffer size1e5 ~ 5e5太大存储开销高存 CPU 即可网络数量2 个 Actor 2 个 CriticTD3 标准配置GPU 利用率监控nvidia-smi避免利用率突增导致 OOM训练前检查清单确认torch.cuda.is_available()为 True设置os.environ[CUDA_VISIBLE_DEVICES] 0所有 tensor 显式调用.to(device)经验回放缓冲区存储 CPU tensor定期执行torch.cuda.empty_cache()使用with torch.no_grad()包裹 target 网络更新 六、监控与调试命令实时显存变化watch-n0.5nvidia-smi记录显存使用历史nvidia-smi --query-gputimestamp,memory.used--formatcsv-l1gpu_mem_log.csvPyTorch 显存分析importtorchprint(torch.cuda.memory_summary())torch.cuda.reset_peak_memory_stats()七、总结查看 GPU 状态熟练使用nvidia-smi及其扩展命令。显存泄漏特征训练初期正常后期 OOM显存占用逐渐增长不回落。修复核心避免计算图残留、存储 CPU tensor、定期清理缓存。RTX 3050 用户控制 batch_size 和网络宽度8GB 显存足够运行中小规模强化学习任务。遇到 OOM 不要慌按照本文步骤排查通常能解决 90% 的问题。如果仍然崩溃请检查是否有其他进程占用显存或考虑使用混合精度训练torch.cuda.amp进一步降低显存占用。 本文基于实际训练日志和错误信息整理而成所有命令已在 Ubuntu 20.04 CUDA 12.2 PyTorch 环境下验证。