如何升级sagemaker中的sklearn库

Question

我注意到我的 Sagemaker (Amazon aws) jupyter 笔记本有一个过时版本的 sklearn 库。

当我运行 ! pip freeze 我得到：

sklearn==0.0

当我运行（python）print(sklearn.__version__)我得到

0.24.1

我不确定哪个是我的真实版本，但我需要 1.0.0 才能使用 from_predictions() 方法。

但是当我尝试运行 ! /home/ec2-user/anaconda3/envs/amazonei_mxnet_p36/bin/python -m pip install --upgrade sklearn 时，我得到以下输出：

Requirement already satisfied: sklearn in /home/ec2-user/anaconda3/envs/amazonei_mxnet_p36/lib/python3.6/site-packages (0.0) Requirement already satisfied: scikit-learn in /home/ec2-user/anaconda3/envs/amazonei_mxnet_p36/lib/python3.6/site-packages (from sklearn) (0.24.1) Requirement already satisfied: scipy>=0.19.1 in /home/ec2-user/anaconda3/envs/amazonei_mxnet_p36/lib/python3.6/site-packages (from scikit-learn->sklearn) (1.5.3) Requirement already satisfied: joblib>=0.11 in /home/ec2-user/anaconda3/envs/amazonei_mxnet_p36/lib/python3.6/site-packages (from scikit-learn->sklearn) (1.0.1) Requirement already satisfied: threadpoolctl>=2.0.0 in /home/ec2-user/anaconda3/envs/amazonei_mxnet_p36/lib/python3.6/site-packages (from scikit-learn->sklearn) (2.1.0) Requirement already satisfied: numpy>=1.13.3 in /home/ec2-user/anaconda3/envs/amazonei_mxnet_p36/lib/python3.6/site-packages (from scikit-learn->sklearn) (1.19.5)

这是一个非常流行的库，所以如果 sagemaker 不能升级它就很奇怪了。有人知道我做错了什么吗？

Answer 1

我设法通过以下命令将 sklearn 更新到版本 0.24.2：

!conda update scikit-learn --yes

要进一步更新它，您可能还需要更新 Python，这是 Sagemaker 当前 conda_python3 内核中的 3.6 版。

创建您的自定义 conda 环境看起来也很有希望，如下所述：https://docs.aws.amazon.com/sagemaker/latest/dg/nbi-add-external.html

如何升级sagemaker中的sklearn库

How to upgrade the sklearn library in sagemaker

python

pip

scikit-learn

amazon-sagemaker