Sagemaker inference on inf1 no opencv

0

I am trying to deploy Pytorch model on ml.inf1.xlarge instance. Image: 301217895009.dkr.ecr.us-west-2.amazonaws.com/sagemaker-neo-pytorch:1.5.1-inf-py3 My python code using some oepncv functions, and when I am trying to run the infernce I got the following error: ModuleNotFoundError: No module named 'cv2'

I tried to add opencv-python-headless to requirements.txt, but then I got another error ImportError: libgthread-2.0.so.0: cannot open shared object file

How I can use opencv with the ml.inf1 instances?

synapse
질문됨 2년 전569회 조회
1개 답변
0

When running Neo inference on Sagemaker, it’s possible to now use the Deep Learning Containers (DLC) provided by AWS. It’s also unnecessary to remain on Pytorch 1.5 when using the DLC images. In our latest released image, OpenCV is pre-installed along with a more recent version of AWS Neuron SDK. Here’s a link to the latest docker image for Pytorch 1.10 + Neuron SDK 1.17: https://github.com/aws/deep-learning-containers/blob/master/pytorch/inference/docker/1.10/py3/sdk1.17.1/Dockerfile.neuron#L83

If you’re unable to easily move to the latest Pytorch+Neuron DLC, the dockerfile link may help with resolving installation errors of OpenCV into your container.

답변함 2년 전
  • I tried different available images (1.5, 1.7, 1.8. 1.9), but results are the same. Error: No module named 'cv2'

    You can easily reproduce the problem, just try to run following code at any "sagemaker-neo-pytorch" images:

    import cv2 print (cv2.version)

    Currently I can't find any new images using following code: from sagemaker import image_uris image_uris.retrieve(framework='inferentia-pytorch',region='us-west-2', version = "1.10" )

로그인하지 않았습니다. 로그인해야 답변을 게시할 수 있습니다.

좋은 답변은 질문에 명확하게 답하고 건설적인 피드백을 제공하며 질문자의 전문적인 성장을 장려합니다.

질문 답변하기에 대한 가이드라인