diff --git a/scripts/install.sh b/scripts/install.sh
index 2f1519a..3bb8059 100755
--- a/scripts/install.sh
+++ b/scripts/install.sh
@@ -10,10 +10,12 @@ if [ -z "${SIA_INSTALL_NO_TRAIN}" ]; then
     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
 fi
 
 if [ -z "${SIA_INSTALL_NO_CORE}" ]; then
     echo "Installing SIA core..."
     python3 -m venv /root/venvs/sia
     /root/venvs/sia/bin/pip install -e /root/sia
+    /root/venvs/sia/bin/ipython kernel install --name=sia
 fi
\ No newline at end of file