AMD support to test.sh and AMD RX 6800 results

ryujaehun / pytorch-gpu-benchmark

Using the famous cnn model in Pytorch, we run benchmarks on various gpu.

MIT License

227 stars 85 forks source link

AMD support to test.sh and AMD RX 6800 results #28

Open lamikr opened 5 months ago

lamikr commented 5 months ago

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 commented 5 months ago

Fixes https://github.com/ryujaehun/pytorch-gpu-benchmark/issues/29