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

多 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