oneapi basekit vars set for pytorch-cpu

Author: sharvil10

intel/runtimes/pytorch/ubi9-python-3.9/Dockerfile

@@ -40,6 +40,8 @@ RUN echo "Installing softwares and packages" && \
     fix-permissions /opt/app-root -P
 
 ENV CPU_ENV=/opt/app-root-cpu
+ENV GPU_ENV=/opt/app-root
+
 COPY --chown=1001:0 Pipfile.lock.cpu ${CPU_ENV}/Pipfile.lock
 
 #CPU env

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

@@ -23,9 +23,10 @@ RUN echo "Installing softwares and packages" && \
     jupyter labextension disable "@jupyterlab/apputils-extension:announcements" && \
     chmod -R g+w /opt/app-root/lib/python3.9/site-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
@@ -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/pytorch-cpu/kernel.json
+
 WORKDIR /opt/app-root/src
 
 ENV JUPYTER_PRELOAD_REPOS="https://github.com/IntelAI/oneAPI-samples"

jupyter/intel/pytorch/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": "pytorch-cpu",
+ "language": "python",
+ "metadata": {
+  "debugger": true
+ }
+}

jupyter/intel/pytorch/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": "pytorch-gpu",
  "language": "python",
  "metadata": {

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

@@ -0,0 +1,5 @@
+#!/bin/bash
+
+source /opt/intel/oneapi/setvars.sh
+
+exec ${JUPYTER_ENV}/bin/python -m ipykernel_launcher $@