多 GPU CUDA 压力测试

多 GPU CUDA 压力测试

💡 原文中文,约700字,阅读约需2分钟。
📝

内容提要

本文介绍了如何进行多GPU的CUDA压力测试,包括下载并解压gpu-burn软件,编译生成可执行文件gpu_burn,默认测试所有GPU,用户可指定测试时间和特定GPU卡。

🎯

关键要点

  • 下载gpu-burn软件:使用命令$ wget https://codeload.github.com/wilicc/gpu-burn/zip/master。
  • 解压缩下载的文件:使用命令$ unzip gpu-burn-master.zip。
  • 进入解压后的目录并编译:确保CUDA环境变量配置成功,使用命令$ cd gpu-burn-master和make进行编译。
  • 编译成功后生成可执行文件gpu_burn:使用命令$ gpu_burn运行测试。
  • 默认测试所有GPU卡:可以通过在命令后添加时间参数来设置测试时长,快速测试设置为100,稳定性测试设置为500。
  • 可以指定特定GPU卡进行测试:使用命令$ CUDA_VISIBLE_DEVICES=0,1 ./gpu_burn 100来指定0号和1号卡。
➡️

继续阅读