install tensorboard if not already installed
@@ -21,5 +21,8 @@ TENSORBOARD_VERSION=$(pip show tensorboard | grep Version | sed "s/[^:]*: *//")
cd ${WORKING_DIR} && test -f "${STARTUP_HOOK}" && source "${STARTUP_HOOK}"
+# Install tensorboard if not already installed (for custom Pytorch images)
+"$DET_PYTHON_EXECUTABLE" -m pip install tensorboard
+
"$DET_PYTHON_EXECUTABLE" -m pip install tensorboard-plugin-profile
exec "$DET_PYTHON_EXECUTABLE" -m determined.exec.tensorboard "$TENSORBOARD_VERSION" "$@"
