Merge pull request #135 from /issues/133/gpu

add setup for desc-python-gpu

Author: heather999

jupyter-kernels/desc-python-gpu/kernel.json

@@ -0,0 +1,10 @@
+{
+  "argv": [
+    "/global/common/software/lsst/common/miniconda/kernels/python-gpu.sh",
+    "-f",
+    "{connection_file}"
+  ],
+  "display_name": "desc-python-gpu",
+  "language": "python"
+}
+

jupyter-kernels/desc-python-gpu/python-gpu.sh

@@ -0,0 +1,9 @@
+#!/bin/bash
+
+INST_DIR=/global/common/software/lsst/common/miniconda
+source $INST_DIR/setup_gpu_python.sh ""
+
+if [ $# -gt 0 ] ; then
+    exec python -m ipykernel $@
+fi
+

jupyter-kernels/setup/setup_gpu_python.sh

@@ -0,0 +1,52 @@
+#!/bin/bash
+
+function log () {
+    if [[ $_V -eq 1 ]]; then
+        echo "$@"
+    fi
+}
+
+_V=0
+while getopts "v" OPTION
+do
+  case $OPTION in
+    v) _V=1
+       ;;
+  esac
+done
+
+export LSST_INST_DIR=/global/common/software/lsst/common/miniconda
+export LSST_PYTHON_VER=gpu
+
+module load PrgEnv-gnu
+module load gpu
+#  module load cpu
+#  module load cudatoolkit
+#  module load cray-mpich-abi/8.1.24
+
+
+unset PYTHONHOME
+unset PYTHONPATH
+export PYTHONNOUSERSITE=' '
+
+export DESC_GCR_SITE='nersc'
+
+source $LSST_INST_DIR/$LSST_PYTHON_VER/etc/profile.d/conda.sh
+conda activate base
+if [ -n "$DESCUSERENV" ]; then
+   conda activate $DESCUSERENV
+fi
+
+if [ -n "$DESCPYTHONPATH" ]; then
+    export PYTHONPATH=$PYTHONPATH:"$DESCPYTHONPATH"
+    echo "Including user python path: $DESCPYTHONPATH"
+fi 
+
+export PYTHONPATH=$PYTHONPATH:$LSST_INST_DIR/$LSST_PYTHON_VER
+
+OUTPUTPY="$(which python)"
+echo Now using "${OUTPUTPY}"
+
+export HDF5_USE_FILE_LOCKING=FALSE
+
+

jupyter-kernels/setup/user_setup.sh

@@ -15,6 +15,7 @@ INST_DIR=/global/common/software/lsst/common/miniconda
 source $INST_DIR/setup_current_python.sh
 jupyter kernelspec install $INST_DIR/kernels/desc-python --user
 jupyter kernelspec install $INST_DIR/kernels/desc-python-dev --user
+jupyter kernelspec install $INST_DIR/kernels/desc-python-gpu --user
 jupyter kernelspec install $INST_DIR/kernels/desc-python-bleed --user
 jupyter kernelspec install $INST_DIR/kernels/desc-python-old --user
 jupyter kernelspec install $INST_DIR/kernels/desc-python-img --user