多 GPU CUDA 压力测试

本文 首发于 🌱 煎茶， 转载 请注明 来源。

多 GPU CUDA 压力测试#

1.下载软件

$ wget https://codeload.github.com/wilicc/gpu-burn/zip/master

2.解压缩

$ unzip gpu-burn-master.zip

3.进入目录编译(确保cuda环境变量已经配置成功 nvcc -v能显示结果)

$ cd gpu-burn-master
make

4.编译成功后,会在当前目录生成 gpu_burn 这个文件

$ gpu_burn

5.默认执行,跑全部GPU卡,空格后面参数为时间,一般快速测试设置100,稳定性测试为500

$ ./gpu_burn 100

6.可以指定某几张卡跑,比如指定0和1号卡

$ CUDA_VISIBLE_DEVICES=0,1 ./gpu_burn 100

References#

• http://wili.cc/blog/gpu-burn.html
• GPU burn 测试gpu
• 如何对 Linux 操作系统的 GPU 实例进行压测？ - 火山引擎-技术服务

注：本作品采用 知识共享署名-非商业性使用-相同方式共享 4.0 国际许可协议 进行许可。