pytorch-gpu-benchmark icon indicating copy to clipboard operation
pytorch-gpu-benchmark copied to clipboard

Published 20 hours ago • verified publisher icon ryujaehun

Test.sh needs changes to allow running tests also for AMD gpus

Open lamikr opened this issue 1 year ago • 0 comments

AMD's rocm platform has rocm-smi tool that is similar to nvidia-smi tool. Add suport to test.sh script to first check the existence of nvidia-smi and nvidia-smi before using them to query the gpu count.

lamikr avatar Jun 07 '24 20:06 lamikr