95 lines
3.4 KiB
Python
Executable File
95 lines
3.4 KiB
Python
Executable File
# Copyright (c) Jupyter Development Team.
|
|
# Distributed under the terms of the Modified BSD License.
|
|
|
|
# Configuration file for JupyterHub
|
|
import os
|
|
|
|
c = get_config() # noqa: F821
|
|
|
|
# We rely on environment variables to configure JupyterHub so that we
|
|
# avoid having to rebuild the JupyterHub container every time we change a
|
|
# configuration parameter.
|
|
|
|
# Spawn single-user servers as Docker containers
|
|
c.JupyterHub.spawner_class = "dockerspawner.DockerSpawner"
|
|
|
|
# Spawn containers from this image
|
|
c.DockerSpawner.image = os.environ["DOCKER_NOTEBOOK_IMAGE"]
|
|
|
|
# Connect containers to this Docker network
|
|
network_name = os.environ["DOCKER_NETWORK_NAME"]
|
|
c.DockerSpawner.use_internal_ip = True
|
|
c.DockerSpawner.network_name = network_name
|
|
|
|
# Explicitly set notebook directory because we'll be mounting a volume to it.
|
|
# Most `jupyter/docker-stacks` *-notebook images run the Notebook server as
|
|
# user `jovyan`, and set the notebook directory to `/home/jovyan/work`.
|
|
# We follow the same convention.
|
|
notebook_dir = os.environ.get("DOCKER_NOTEBOOK_DIR", "/home/jovyan/work")
|
|
c.DockerSpawner.notebook_dir = notebook_dir
|
|
|
|
# Mount the real user's Docker volume on the host to the notebook user's
|
|
# notebook directory in the container
|
|
# c.DockerSpawner.volumes = {"jupyterhub-user-{username}": notebook_dir}
|
|
# Mount the real user's Docker volume on the host to the notebook user's
|
|
# notebook directory in the container
|
|
c.DockerSpawner.volumes = {
|
|
"jupyterhub-user-{username}": notebook_dir,
|
|
"/mnt/mydrive": "/home/jovyan/work/mydrive",
|
|
"/mnt/mydrive/project/docker-jupyterhub/id_rsa": "/home/jovyan/.ssh/id_rsa",
|
|
}
|
|
|
|
|
|
# Remove containers once they are stopped
|
|
c.DockerSpawner.remove = True
|
|
|
|
# For debugging arguments passed to spawned containers
|
|
c.DockerSpawner.debug = True
|
|
# c.Application.log_level = 'DEBUG'
|
|
|
|
# User containers will access hub by container name on the Docker network
|
|
c.JupyterHub.hub_ip = 'jupyterhub'
|
|
c.JupyterHub.hub_port = 8080
|
|
|
|
# Persist hub data on volume mounted inside container
|
|
c.JupyterHub.cookie_secret_file = "/data/jupyterhub_cookie_secret"
|
|
c.JupyterHub.db_url = "sqlite:////data/jupyterhub.sqlite"
|
|
|
|
# Authenticate users with Native Authenticator
|
|
c.JupyterHub.authenticator_class = "nativeauthenticator.NativeAuthenticator"
|
|
|
|
# Allow anyone to sign-up without approval
|
|
c.NativeAuthenticator.open_signup = True
|
|
|
|
# Allowed admins
|
|
admin = os.environ.get("JUPYTERHUB_ADMIN")
|
|
if admin:
|
|
c.Authenticator.admin_users = [admin]
|
|
|
|
# c.DockerSpawner.extra_create_kwargs.update({
|
|
# "environment": {"JUPYTER_ENABLE_LAB": "yes"}
|
|
# })
|
|
|
|
# 启动jupyter时候增加跨域支持, 否则反向代理的时候出现问题
|
|
# --NotebookApp.iopub_data_rate_limit=10000000 给nglview使用
|
|
c.DockerSpawner.extra_create_kwargs.update({
|
|
"environment": {"NOTEBOOK_ARGS": "--NotebookApp.allow_origin='*' --NotebookApp.iopub_data_rate_limit=10000000"}
|
|
})
|
|
|
|
|
|
# 要支持正则匹配的域名请求，可以通过设置 allow_origin_pat 参数来实现。这个参数允许你使用正则表达式来匹配允许跨域请求的域名。例如，如果你想允许所有以 .example.com 结尾的域名进行跨域请求，可以在 jupyterhub_config.py 文件中添加如下配置：
|
|
|
|
# c.Spawner.environment = {
|
|
# 'JUPYTERHUB_CORS': '{"allow_origin_pat": "https?://.*\\.example\\.com"}'
|
|
# }
|
|
|
|
# GPU 和网络配置
|
|
c.DockerSpawner.extra_host_config = {
|
|
'runtime': 'nvidia'
|
|
}
|
|
c.DockerSpawner.environment = {
|
|
'NVIDIA_DRIVER_CAPABILITIES': 'compute,utility',
|
|
'NVIDIA_VISIBLE_DEVICES': 'all'
|
|
}
|
|
|