colab在更换python包版本时，如何正确重启

Posted 2023-01-28

当我在使用Node2Vec这个包的时候，遇到了gem和numpy的版本冲突问题。最后在pc上测试，发现只要升级numpy就能解决问题。但是在colab中更新numpy版本依旧报错。后面才发现，就算在colab更新numpy，colab并不会去刷新python包的注册表。所以需要我们去重启colab。网上有很多方法，我只给出我测试正确的方法：

try:
  from gem.utils import graph_util, plot_util
except (ImportError, KeyError, ModuleNotFoundError):
  ## code to install gem
  print(Stopping RUNTIME. Colaboratory will restart automatically. Please run again.)
  exit()

把这个代码放上去，运行一下。再用下面这个代码查看自己调用的是否是想要的版本即可。

import numpy as np
print(np.__version__)