gpu压力测试工具----gpu_burn的使用

本文主要是介绍gpu压力测试工具----gpu_burn的使用，希望对大家解决编程问题提供一定的参考价值，需要的开发者们随着小编来一起学习吧！

背景：

        我们平时想做gpu的压力测试，需要一个简单的可以一键执行的工具来测试，主要要来打满gpu的使用率和显存，这里以8卡4090机器为测试机，驱动版本550.54.14，cuda版本12.3。

开源方案：

gpu-burn一个开源的cuda测试工具，一个专门用来给 NVIDIA GPU 施加压力的工具。

测试方法：

（1）物理服务器或者容器里执行GPU压力测试

# 克隆仓库
git clone https://github.com/wilicc/gpu-burn.git# 进入目录
cd gpu-burn# 编译
make# 将 gpu_burn 程序运行在后台，并将输出重定向到 output.log 文件
./gpu_burn 60 > output.log 2>&1 &

未运行压测前GPU使用概览：

运行之后：

可以看到GPU的利用率和显存基本都被打满

（2）以容器方式进行测试

git clone https://github.com/wilicc/gpu-burn
cd gpu-burn
docker build -t gpu_burn .
docker run --rm --gpus all gpu_burn

这篇关于gpu压力测试工具----gpu_burn的使用的文章就介绍到这儿，希望我们推荐的文章对编程师们有所帮助！

---