Replace deprecated Python dependency sklearn with scikit-learn. (#13585)

This commit is contained in:
Edward Chen 2022-11-08 09:08:29 -08:00 committed by GitHub
parent efcbdac58e
commit 9e65f3bfdb
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
6 changed files with 7 additions and 8 deletions

View file

@ -1,6 +1,6 @@
# docker build --network=host --file Dockerfile.rocm4.3.1.pytorch --tag ort:rocm4.3.1-pytorch .
FROM rocm/pytorch:rocm4.3.1_ubuntu18.04_py3.6_pytorch_1.9.0
FROM rocm/pytorch:rocm4.3.1_ubuntu18.04_py3.6_pytorch_1.9.0
RUN apt-get -y install gpg-agent
RUN wget -q -O - http://repo.radeon.com/rocm/rocm.gpg.key | apt-key add -
@ -137,7 +137,7 @@ RUN git clone -b wezhan/tnlrv4 --recursive https://github.com/microsoft/onnxrunt
RUN pip3 install --no-cache-dir GPUtil azureml azureml-core datasets tokenizers ninja cerberus sympy sacremoses sacrebleu
RUN pip install transformers==2.10.0 sklearn tensorboardX
RUN pip install transformers==2.10.0 scikit-learn tensorboardX
RUN pip install --pre torch-ort -f https://download.onnxruntime.ai/torch_ort_nightly.html
RUN python -m torch_ort.configure

View file

@ -1,3 +1,3 @@
sklearn
scikit-learn
transformers==v4.4.2
wget

View file

@ -4,7 +4,7 @@ torch==1.12.0
setuptools>=41.4.0
cerberus
h5py
sklearn
scikit-learn
numpy
pandas
parameterized

View file

@ -1,5 +1,5 @@
pandas
sklearn
scikit-learn
numpy==1.21.6
transformers==v4.4.2
tensorboard>=2.2.0,<2.5.0

View file

@ -1,7 +1,7 @@
--pre
-f https://download.pytorch.org/whl/torch_stable.html
# transformers requires sklearn
sklearn
# transformers requires scikit-learn
scikit-learn
numpy==1.21.6
transformers==v2.10.0
torch==1.10.0+cu113

View file

@ -33,7 +33,6 @@ RUN pip install \
sacremoses \
scipy \
scikit-learn \
sklearn \
tokenizers \
sentencepiece \
dill==0.3.4 \