如果在 ! 在 Google Colab 中安装 pip？

Question

在 Google Colab 中，当我在每次运行时安装并导入此代码时：

! pip install --upgrade snowflake-connector-python
! pip install pyspark
! pip install snowflake-connector-python asn1crypto==0.24.0
! pip install azure-storage==0.34.3
! pip install azure.storage.common

from pyspark.sql import SparkSession,SQLContext
import pyspark
import snowflake.connector as con
from pyspark.sql import SQLContext
sc = pyspark.SparkContext.getOrCreate()
sqlContext = SQLContext(sc)

我收到此错误：

KeyError: 'snowflake-connector-python'

我已经和 Snowflake 谈过了，他们说这是正常的，我只需要再次单击“重新启动并全部运行”即可。

我想知道是否有一种方法可以在代码中自动重新启动并运行所有，这样我就不需要手动执行此操作？ 我在想一个 while 循环，但我不熟悉安装过程中的错误处理。

我知道我可以将 pip 安装保存到我的云端硬盘，但由于这些文件将充当共享仪表板，我不希望同事必须这样做。

我已经阅读了一些关于如何重新启动内核，但不再运行整个文件的内容，例如：

import os
os._exit(00)

Answer 1

这是我能做的最好的事情：

• 尝试导入一个包
• 如果失败（KeyError、ImportError 等），pip install 并强制重启内核。

try:
  import snowflake.connector as con
except (ImportError, KeyError, ModuleNotFoundError):
  !pip install snowflake-connector-python 
  print('Stopping RUNTIME. Colaboratory will restart automatically. Please run again.')
  exit()

基于对Google Colab 的多个答案- 如何使用 python 代码或命令行界面“重新启动运行时”？ .