AMD GPU Benchmark

Ubuntu 22.04

Install docker :

Install AMD TF-ROCm Image

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/ --num_1 --model resnet50 --batch_size 32

total images/sec: 477.72

Leave a Reply

Your email address will not be published. Required fields are marked *