diff --git a/build/test-requirements.txt b/build/test-requirements.txt
index 31e8b8bb1be8..03033b7d489b 100644
--- a/build/test-requirements.txt
+++ b/build/test-requirements.txt
@@ -18,6 +18,7 @@ isort
 
 # Integrated TensorBoard tests
 tensorboard ; python_version > '2.7'
+torch-tb-profiler ; python_version > '2.7'
 
 # Python 2.7 support.
 pytest==4.6.9 ; python_version == '2.7'