Skip to content

Commit

Permalink
ask user to install tf-cpu instead for the profiler (#7212)
Browse files Browse the repository at this point in the history
  • Loading branch information
JackCaoG committed Jun 6, 2024
1 parent 8748a2f commit 26327a2
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion examples/debug/train_resnet_profile.py
Original file line number Diff line number Diff line change
Expand Up @@ -25,6 +25,6 @@
f'localhost:{profile_port}', profile_logdir, duration_ms=duration_ms)
base.start_training()
# You can view the profile at tensorboard by
# 1. pip install tensorflow tensorboard-plugin-profile
# 1. pip install tensorflow-cpu tensorboard-plugin-profile
# 2. tensorboard --logdir /tmp/profile/ --port 6006
# For more detail plase take a look at https://cloud.google.com/tpu/docs/pytorch-xla-performance-profiling-tpu-vm

0 comments on commit 26327a2

Please sign in to comment.