问题描述 #

最近Google Colab默认的Numpy包升级到了2.0.2，这导致我的部分Python代码因为兼容性问题无法正常执行。我需要使用Numpy 1.26.4，于是在Colab中执行了：

!pip install numpy==1.26.4

安装日志显示已成功卸载Numpy-2.0.2并安装了Numpy-1.26.4，但代码依然报错。通过以下代码检查版本：

import numpy
numpy.version.version

发现Colab仍在使用2.0.2版本。直到我手动重启会话后，版本才切换为1.26.4，代码也能正常运行了。

但我没法每次运行笔记本都手动重启会话再执行后续代码，我找到了程序化重启会话的命令：

import os
os.kill(os.getpid(), 9)

但不知道重启后如何自动继续执行下面单元格的代码。现在我需要解决至少一个以下问题：

1. 有没有办法安装Numpy 1.26.4后无需重启会话就让它生效？
2. 有没有办法在会话程序化重启后，自动继续执行后续单元格的代码？
3. 能不能把Numpy 1.26.4存到虚拟环境或Drive中，每次加载直接使用它而不是Colab默认的2.0.2？

解决方案 #

针对问题1：无需重启让Numpy版本生效 #

当前会话中旧版本的Numpy模块可能仍在内存中加载，即使安装了新版本也不会立即切换。你可以尝试以下两种方法：

方法1：用importlib重新加载Numpy

!pip install numpy==1.26.4
import importlib
import numpy
# 重新加载Numpy模块
importlib.reload(numpy)
# 检查版本
print(numpy.version.version)

注意：这种方法不一定总能生效，如果其他已导入模块依赖旧版本Numpy，可能会出现冲突。

方法2：删除已加载的Numpy模块后重新导入

!pip install numpy==1.26.4
import sys
# 移除已加载的Numpy模块
if 'numpy' in sys.modules:
    del sys.modules['numpy']
# 重新导入Numpy
import numpy
print(numpy.version.version)

如果以上方法仍无法切换版本，说明模块依赖绑定过深，还是需要依赖会话重启。

针对问题2：程序化重启后自动继续执行代码 #

你可以通过设置本地标记文件让笔记本在重启后自动检测标记并继续执行后续代码，具体步骤如下：

步骤1：安装Numpy并设置重启标记

在安装Numpy的单元格中加入以下代码：

import numpy as np
# 检查当前版本是否为目标版本
if np.version.version != "1.26.4":
    !pip install numpy==1.26.4
    # 创建标记文件，说明重启后需要继续执行
    with open('/content/continue_flag.txt', 'w') as f:
        f.write('1')
    # 程序化重启会话
    import os
    os.kill(os.getpid(), 9)
else:
    # 若已为目标版本，清理标记文件（如果存在）
    import os
    if os.path.exists('/content/continue_flag.txt'):
        os.remove('/content/continue_flag.txt')

步骤2：后续单元格添加标记检测

在所有需要自动执行的后续单元格开头，加入以下检测代码：

import os
# 检查是否存在继续执行的标记
if os.path.exists('/content/continue_flag.txt'):
    # 清理标记（避免重复执行）
    os.remove('/content/continue_flag.txt')
    # 这里写当前单元格的业务代码
    # 示例：
    print("会话重启完成，开始执行后续代码...")
    # 你的其他代码逻辑...

更高效的批量执行方案

如果后续代码较多，可以把所有业务代码存为单独的脚本，重启后自动运行：

# 安装Numpy的单元格
!pip install numpy==1.26.4
# 把后续业务代码写入脚本
%%writefile /content/business_code.py
import numpy as np
print("当前Numpy版本：", np.version.version)
# 这里写入所有后续需要执行的代码...

# 创建重启标记
with open('/content/run_business.txt', 'w') as f:
    f.write('1')
# 重启会话
import os
os.kill(os.getpid(), 9)

然后在笔记本的第一个单元格加入以下初始化代码：

import os
from google.colab import drive  # 如果用到Drive的话

# 检查是否需要运行业务代码
if os.path.exists('/content/run_business.txt'):
    os.remove('/content/run_business.txt')
    # 自动运行业务脚本
    %run /content/business_code.py
else:
    print("首次执行，开始安装指定版本Numpy...")

针对问题3：将Numpy 1.26.4存到Google Drive并加载使用 #

你可以把Numpy 1.26.4安装到Google Drive的自定义路径，每次运行笔记本时直接从该路径加载，无需重复安装或重启会话：

步骤1：挂载Google Drive并安装Numpy到自定义路径

from google.colab import drive
drive.mount('/content/drive')
# 安装Numpy到Drive的自定义目录
!pip install numpy==1.26.4 --target=/content/drive/MyDrive/colab_custom_libs/

步骤2：每次运行笔记本时加载自定义路径的Numpy

from google.colab import drive
drive.mount('/content/drive')
import sys
# 将自定义包路径添加到Python模块搜索路径的最前面
sys.path.insert(0, '/content/drive/MyDrive/colab_custom_libs/')
# 导入Numpy
import numpy
print(numpy.version.version)

这种方法直接绕过Colab的默认包，每次都从Drive的自定义路径加载指定版本的Numpy，无需重启或重复安装。

备注：内容来源于stack exchange，提问作者Ben Bost