AMD GPU Benchmark

Ubuntu 22.04

Install docker :

https://www.digitalocean.com/community/tutorials/how-to-install-and-use-docker-on-ubuntu-22-04

Install AMD TF-ROCm Image

https://rocm.docs.amd.com/en/latest/how_to/tensorflow_install/tensorflow_install.html#../../deploy/linux/install.md

docker pull rocm/tensorflow:latest

Run it and enter the docker console

docker run -it --network=host --device=/dev/kfd --device=/dev/dri \
--ipc=host --shm-size 16G --group-add video --cap-add=SYS_PTRACE \
--security-opt seccomp=unconfined rocm/tensorflow:latest

Do benchmarking

python3 benchmarks/scripts/tf_cnn_benchmarks/tf_cnn_benchmarks.py --num_1 --model resnet50 --batch_size 32

total images/sec: 477.72