fix: issue with install and version check of tensorboard
A different version of tensorboard could be installed after we check the
version.
@@ -15,8 +15,6 @@ if ! /bin/which "$DET_PYTHON_EXECUTABLE" >/dev/null 2>&1 ; then
exit 1
fi
-TENSORBOARD_VERSION=$(pip show tensorboard | grep Version | sed "s/[^:]*: *//")
-
"$DET_PYTHON_EXECUTABLE" -m pip install -q --user /opt/determined/wheels/determined*.whl
cd ${WORKING_DIR} && test -f "${STARTUP_HOOK}" && source "${STARTUP_HOOK}"
@@ -24,5 +22,7 @@ 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
+TENSORBOARD_VERSION=$(pip show tensorboard | grep Version | sed "s/[^:]*: *//")
+
"$DET_PYTHON_EXECUTABLE" -m pip install tensorboard-plugin-profile
exec "$DET_PYTHON_EXECUTABLE" -m determined.exec.tensorboard "$TENSORBOARD_VERSION" "$@"
