def JupyterLabPlugin():
return PluginConfiguration(
name="jupyterlab",
ports=[PluginPort(internal=8889, public=True)],
target_role=PluginTargetRole.All,
execute="jupyter_lab.sh",
files=[PluginFile("jupyter_lab.sh", os.path.join(dir_path, "jupyter_lab.sh"))],
)
def InstallPlugin(name, command, packages=None):
return PluginConfiguration(
name=name,
target_role=PluginTargetRole.All,
execute="install.sh",
files=[PluginFile("install.sh", os.path.join(dir_path, "install.sh"))],
args=packages,
env=dict(COMMAND=command))
def NvBLASPlugin():
return PluginConfiguration(
name="nvblas",
ports=[],
target_role=PluginTargetRole.All,
execute="nvblas.sh",
files=[PluginFile("nvblas.sh", os.path.join(dir_path, "nvblas.sh"))],
)
def TensorflowOnSparkPlugin():
return PluginConfiguration(
name="tensorflow_on_spark",
target_role=PluginTargetRole.Master,
execute="tensorflow_on_spark.sh",
files=[
PluginFile("tensorflow_on_spark.sh", os.path.join(dir_path, "tensorflow_on_spark.sh")),
],
)
def RStudioServerPlugin(version="1.1.383"):
return PluginConfiguration(
name="rstudio_server",
ports=[PluginPort(internal=8787, public=True)],
target_role=PluginTargetRole.Master,
execute="rstudio_server.sh",
files=[
PluginFile("rstudio_server.sh",
os.path.join(dir_path, "rstudio_server.sh"))
],
env=dict(RSTUDIO_SERVER_VERSION=version),
)