oneapi basekit vars set for tensorflow-cpu

Author: sharvil10

jupyter/intel/tensorflow/ubi9-python-3.9/Dockerfile

@@ -25,7 +25,7 @@ RUN echo "Installing softwares and packages" && \
     fix-permissions /opt/app-root -P
 
 #Replacing kernel manually with oneapi variable setting script
-COPY --chown=1001:0 start-gpu-kernel /opt/app-root/bin/start-gpu-kernel
+COPY --chown=1001:0 start-kernel /opt/app-root/bin/start-kernel
 COPY --chown=1001:0 kernel-gpu.json /opt/app-root/share/jupyter/kernels/python3/kernel.json
 COPY --chown=1001:0 start-notebook.sh /opt/app-root/bin
 COPY --chown=1001:0 builder /opt/app-root/builder
@@ -35,6 +35,7 @@ COPY --chown=1001:0 utils /opt/app-root/bin/utils
 COPY --chown=1001:0 Pipfile.lock.cpu ${CPU_ENV}/Pipfile.lock
 
 WORKDIR ${CPU_ENV}
+
 RUN source ${CPU_ENV}/bin/activate && \
     micropipenv install && \
     rm -f ./Pipfile.lock && \
@@ -45,6 +46,8 @@ RUN source ${CPU_ENV}/bin/activate && \
     fix-permissions ${CPU_ENV} -P && \
     chmod -R g+w /opt/app-root/src
 
+COPY --chown=1001:0 kernel-cpu.json /opt/app-root/share/jupyter/kernels/tensorflow-cpu/kernel.json
+
 WORKDIR /opt/app-root/src
 
 ENV JUPYTER_PRELOAD_REPOS="https://github.com/IntelAI/oneAPI-samples"

jupyter/intel/tensorflow/ubi9-python-3.9/kernel-cpu.json

@@ -0,0 +1,13 @@
+{
+ "argv": [
+  "/opt/app-root/bin/start-kernel",
+  "-f",
+  "{connection_file}"
+ ],
+ "env": {"JUPYTER_ENV":"/opt/app-root-cpu"},
+ "display_name": "tensorflow-cpu",
+ "language": "python",
+ "metadata": {
+  "debugger": true
+ }
+}

jupyter/intel/tensorflow/ubi9-python-3.9/kernel-gpu.json

@@ -1,9 +1,10 @@
 {
  "argv": [
-  "/opt/app-root/bin/start-gpu-kernel",
+  "/opt/app-root/bin/start-kernel",
   "-f",
   "{connection_file}"
  ],
+ "env": {"JUPYTER_ENV":"/opt/app-root"},
  "display_name": "tensorflow-gpu",
  "language": "python",
  "metadata": {

jupyter/intel/tensorflow/ubi9-python-3.9/start-kernel

@@ -2,4 +2,4 @@
 
 source /opt/intel/oneapi/setvars.sh
 
-exec /opt/app-root/bin/python -m ipykernel_launcher $@
+exec ${JUPYTER_ENV}/bin/python -m ipykernel_launcher $@