How to use albumentations in in AWS Sagemaker?

Hello! I am new to Huggingface and Sagemaker, and I am trying to fine-tune a Huggingface model in Sagemaker using the HuggingFace Estimator.

In my file, I need to import and use the albumentations library for data augmentation. I have already installed albumentations in Sagemaker notebook using pip install. However, when I run, it cannot recognize that the albumentations library is installed.

I have looked for solutions to this issue in the forums, but haven’t been able to find a solution that works for me. Can anyone please help me with this issue? Thanks in advance!

Similar question in the forum fyi: