如何在创建 ai 平台笔记本时安装 python 库

时间:2021-05-29 01:46:57

标签: python google-cloud-platform script gcp-ai-platform-notebook

我想在 GCP 中创建笔记本实例时使用“选择创建后运行的脚本”。

具体来说,我想用它来安装python包。

我需要写什么样的脚本(扩展和内容)?

1 个答案:

答案 0 :(得分:0)

这将是安装 Voila 的启动后脚本示例。 将此文件保存在 GCS 存储桶中,并在创建 Notebook 时定义路径,例如:

gcloud notebooks instances create nb-1 \
'--vm-image-project=deeplearning-platform-release' \
'--vm-image-family=tf2-latest-cpu' \
'--metadata=post-startup-script=gs://ai-platform-notebooks-tools/install-voila.sh' \
'--location=us-central1-a'

脚本内容:

#!/bin/bash -eu
# Installs Voila in AI Platform Notebook

function install_voila() {
  echo 'Installing voila...'
  /opt/conda/condabin/conda install -y -c conda-forge ipywidgets ipyvolume bqplot scipy
  /opt/conda/condabin/conda install -y -c conda-forge voila
  /opt/conda/bin/jupyter lab build
  systemctl restart jupyter.service || echo 'Error restarting jupyter.service.'
}

function download_samples() {
  echo 'Downloading samples...'
  cd /home/jupyter
  git clone https://github.com/voila-dashboards/voila

}

function main() {
  install_voila || echo 'Error installing voila.' 
  download_samples || echo 'Error downloading voila samples.' 
}

main