Originally furnished by Min Dong, 2019-03-01 14:30 EST.master
parent
28eb7a0d98
commit
1c8a5da492
1 changed files with 52 additions and 0 deletions
@ -0,0 +1,52 @@ |
||||
#!/bin/bash -l |
||||
|
||||
jupyter_output=$(mktemp ~/.jupyter_output.XXXX) |
||||
|
||||
clean_up() { |
||||
job_id=$(awk '/srun: job/ {print $3;exit}' $jupyter_output) |
||||
scancel $job_id |
||||
rm $jupyter_output |
||||
} |
||||
|
||||
trap clean_up EXIT |
||||
|
||||
set -m |
||||
|
||||
|
||||
if [ x"$PYTHON_VER" = x ]; then |
||||
echo "Please load python and any needed python modules before launching juypter!" |
||||
echo "Example: " |
||||
echo " enable_lmod" |
||||
echo " module load python/3.6 # 2.7/3.7 is also avaiable" |
||||
exit 1 |
||||
fi |
||||
|
||||
echo " * loading jupyter module" |
||||
|
||||
enable_lmod |
||||
module load python/$PYTHON_VER |
||||
module load jupyter |
||||
|
||||
echo " * launching jupyter notebook server" |
||||
(srun -n 1 -c 2 -t 240 -J juypter jupyter-notebook --no-browser --port=$((8000 + $RANDOM % 1000)) --ip=0.0.0.0 2>&1 | tee $jupyter_output > /dev/null) & |
||||
|
||||
echo |
||||
echo " jupyter server will be on for 4 hour" |
||||
echo " please do not close this window, or jupyter will be terminated immediately" |
||||
echo |
||||
|
||||
|
||||
sleep 10 |
||||
|
||||
job_id=$(awk '/srun: job/ {print $3;exit}' $jupyter_output) |
||||
tokens=$(awk -F: '/ http:/ {print $3}' $jupyter_output) |
||||
host=$(squeue -j $job_id -O NodeList -h | sed 's/ //g') |
||||
url="http://$host:$tokens" |
||||
|
||||
echo " * launching browser" |
||||
(firefox $url 2>&1 )>/dev/null & |
||||
|
||||
echo " if you closed browser, please use below to connect back" |
||||
echo " $url" |
||||
|
||||
wait |
Loading…
Reference in new issue