How to use albumentations in train.py 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.https://sagemaker.readthedocs.io/en/stable/frameworks/huggingface/sagemaker.huggingface.htmlIn mytrain.pyfile, I need to import and use thealbumentationslibrary for data augmentation. I have already installedalbumentationsin Sagemaker notebook using pip install. However, when I runtrain.py, it cannot recognize that thealbumentationslibrary 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:Setting up environment in Sagemaker StudioAmazon SageMakerHello everyone, 
I am currently working on finetuning a DETR object detection model in Sagemaker Studio using a Hugging Face Estimator. I have installed the libraries with pip, as shown below: 
!pip install transformers datasets huggingface_hub evaluate timm albumentations wandb sagemaker

However, when I tried to check the versions of Pytorch and transformers inside the SageMaker Studio notebook, I got the following error: 
import torch
print(torch.__version__)
>>>ModuleNotFoundError: No module…