1 Answer
- Newest
- Most votes
- Most comments
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.
answered 2 years ago
Relevant content
- asked 9 months ago
- asked 2 days ago
- AWS OFFICIALUpdated 7 months ago
- AWS OFFICIALUpdated 2 years ago
- AWS OFFICIALUpdated 2 years ago
- AWS OFFICIALUpdated a year ago
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" )