python - Google Colab 无法更改 Python 版本

Question

最近，我所有的笔记本都已更新到 python 3.10，我无法运行我的旧代码。我使用的是 jax 0.2.17 和 numpyro 0.7.1，但这些不再适用于我的版本，所以我尝试更改为 python 3.9。我使用了以下代码:

!sudo apt-get update -y

!sudo apt-get install python3.9

!sudo update-alternatives --install /usr/bin/python3 python3 /usr/bin/python3.9 1

!sudo update-alternatives --config python3

!apt-get install python3-pip

但是，下载此文件并使用 !pip install jax==0.2.17 后，当我使用 jax.__version__ 检查 jax 版本时>，我得到 0.4.8，这是 Google Colab 中已安装的 jax 版本。有什么方法可以解决这个问题吗？我希望能够使用旧版本的 jax 和 numpyro ，并且 pip 安装它们，但是当我导入时，我似乎是从 3.10 版本导入的。提前致谢!

Answer 1

Colab 运行时仅支持单个 Python 版本，通常无法安装不同的版本。 Colab 进行 Python 版本升级后，通常会在短时间内提供回退，目前将在接下来的几周内提供 Python 3.9 回退(详细信息请参阅 Colab Updated to Python 3.10)。

然而，几周后，此后备将不再可用，并且无法将 Python 3.9 与 Colab 一起使用。鉴于此，我建议升级您的代码以与更新的软件包版本兼容(我相信 JAX v0.3.11 或更高版本支持 Python 3.10)。如果或由于某种原因这是不可能的，那么您可能会考虑在本地 Jupyter Notebook 运行时中运行代码，这样您可以对计算环境进行更细粒度的控制。