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

Published 20 hours ago • verified publisher icon ryujaehun

AMD support to test.sh and AMD RX 6800 results

Open lamikr opened this issue 1 year ago • 1 comments

Modify test.sh so that both the AMD and Nvidia GPU's are supported for running benchmarks. Add also the csv results for AMD RX 6800. Tested on pytorch 2.3.0 and pytorch vision 0.18.0 on rocm 6.1.1 platform.

lamikr avatar Jun 07 '24 20:06 lamikr