Skip to content

Instantly share code, notes, and snippets.

@nikAizuddin

nikAizuddin/README.md

Created September 23, 2023 17:06
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save nikAizuddin/238d773a6aa3a3b3ce2e7650b79f6908 to your computer and use it in GitHub Desktop.
Save nikAizuddin/238d773a6aa3a3b3ce2e7650b79f6908 to your computer and use it in GitHub Desktop.
Download ZIP
ROCM Tensorflow
Raw
README.md

Spawn:

podman run --rm -it --network=host \
  --device=/dev/kfd \
  --device=/dev/dri \
  --ipc=host \
  --security-opt label=disable \
  docker.io/rocm/tensorflow:rocm5.5-tf2.11-dev \
  bash

Benchmark:

cd /root/benchmarks/scripts/tf_cnn_benchmarks
TF_ROCM_FUSION_ENABLE=1 python3 tf_cnn_benchmarks.py --num_gpus=1 --batch_size=32 --model=resnet50 --use_fp16
TF_ROCM_FUSION_ENABLE=1 python3 tf_cnn_benchmarks.py --num_gpus=1 --batch_size=64 --model=resnet50
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment