eaaeefce55
* add a basic notebook for lab users in the root * update notebook and fix cors for jupyter * cell is code * fix eval batch size check * remove intro notebook
41 lines
1.2 KiB
Bash
Executable File
41 lines
1.2 KiB
Bash
Executable File
#!/bin/bash
|
|
|
|
# Export specific ENV variables to /etc/rp_environment
|
|
echo "Exporting environment variables..."
|
|
printenv | grep -E '^RUNPOD_|^PATH=|^_=' | sed 's/^\(.*\)=\(.*\)$/export \1="\2"/' >> /etc/rp_environment
|
|
echo 'source /etc/rp_environment' >> ~/.bashrc
|
|
|
|
if [[ $PUBLIC_KEY ]]; then
|
|
# runpod
|
|
mkdir -p ~/.ssh
|
|
chmod 700 ~/.ssh
|
|
echo $PUBLIC_KEY >> ~/.ssh/authorized_keys
|
|
chmod 700 -R ~/.ssh
|
|
# Start the SSH service in the background
|
|
service ssh start
|
|
elif [ -n "$SSH_KEY" ]; then
|
|
# latitude.sh
|
|
mkdir -p ~/.ssh
|
|
chmod 700 ~/.ssh
|
|
echo $SSH_KEY >> ~/.ssh/authorized_keys
|
|
chmod 700 -R ~/.ssh
|
|
# Start the SSH service in the background
|
|
service ssh start
|
|
else
|
|
echo "No PUBLIC_KEY or SSH_KEY environment variable provided, not starting openSSH daemon"
|
|
fi
|
|
|
|
# Check if JUPYTER_PASSWORD is set and not empty
|
|
if [ -n "$JUPYTER_PASSWORD" ]; then
|
|
# Set JUPYTER_TOKEN to the value of JUPYTER_PASSWORD
|
|
export JUPYTER_TOKEN="$JUPYTER_PASSWORD"
|
|
fi
|
|
|
|
if [ "$JUPYTER_DISABLE" != "1" ]; then
|
|
# Run Jupyter Lab in the background
|
|
jupyter lab --port=8888 --ip=* --allow-root --ServerApp.allow_origin=* --ServerApp.preferred_dir=/workspace &
|
|
fi
|
|
|
|
# Execute the passed arguments (CMD)
|
|
exec "$@"
|