In container: pytorch import library error

Hi,guys:
I get the following error when using the sklearn library in the container “nvcr.io/nvidia/pytorch:21.03-py3”:

<ModuleNotFoundError: No module named ‘sklearn.metrics.base’>

I looked for a solution in Google but failed.Any idea?Thansk.