Add support for jupyter notebooks for training

2025-03-28 16:10:43 +00:00
parent 468ef2ae1f
commit 56242496bb
5 changed files with 343 additions and 11 deletions
--- a/scripts/install.sh
+++ b/scripts/install.sh
@@ -9,6 +9,7 @@ source "/etc/profile.d/venv_itb.sh"
 echo "Installing Train tool..."
 python3 -m venv "/root/venvs/train"
 /root/venvs/train/bin/pip install -e /root/sia/tools/train/
+/root/venvs/train/bin/ipython kernel install --name=train
 echo "PATH=\"/root/venvs/train/bin/:\$PATH\"" > "/etc/profile.d/venv_train.sh"
 source "/etc/profile.d/venv_train.sh"