code: def install_pkg(path, pkg, base="fw/"):
      if pkg.split("==")[0] if "==" in pkg else pkg == "torch":
          subprocess.run(
              f"pip3 install --upgrade {pkg} --target {path} --default-timeout=100"
              " --extra-index-url https://download.pytorch.org/whl/cpu "
              " --no-cache-dir",
              shell=True,
          )
      elif pkg.split("==")[0] == "jax":
          subprocess.run(
              f"pip install --upgrade {pkg} --target  {path}  -f"
              " https://storage.googleapis.com/jax-releases/jax_releases.html  "
              " --no-cache-dir",
              shell=True,
          )
      else:
          subprocess.run(
              f"pip3 install --upgrade {pkg} --target {path} --default-timeout=100  "
              " --no-cache-dir",
              shell=True,
          )
  
imports: import os
  import subprocess
  import sys
  import json
label_desc: Use image node_id 2 for calling a global function with example usage: install_pkg(path, pkg, base) without return types