我的小小AI 天地

小編最近買了一張nvidia-3070ti顯卡回來，一裝機馬上就想測測看GPU有沒有問題，上網找了一下有一套軟體GPU_burn 專門做燒機測試，也提供docker版本，若不知道如何安裝nvidia-docker 的可以參考小編之前的文章
快速安裝深度學習系統 Ubuntu20.04 tensorflow 2.7.0 docker

https://github.com/wilicc/gpu-burn
如果裝好docker的話，到上述網站把程式碼載下來然後按照說明跑指令就可以測試囉~

step1

git clone https://github.com/wilicc/gpu-burn

step2

cd gpu-burn

step3

docker build -t gpu_burn .

step4

docker run --rm --gpus all gpu_burn

成功執行完的結果

如果想要執行燒機久一點的話可以到Dockerfile 檔案裡面把數字改大一點，預測是測試1分鐘

FROM nvidia/cuda:11.1.1-devel AS builder

WORKDIR /build

COPY . /build/

RUN make

FROM nvidia/cuda:11.1.1-runtime

COPY --from=builder /build/gpu_burn /app/
COPY --from=builder /build/compare.ptx /app/

WORKDIR /app

## 可以把60改成想要的秒數
CMD ["./gpu_burn", "60"]

https://blog.gtwang.org/linux/linux-nvidia-gpu-benchmarking-and-stress-testing/