Deploy SAM2.1 with Sagemaker

Inference Endpoints on the Hub
1

I am trying to deploy SAM2 with this the lasts image to sagemaker :

763104351884.dkr.ecr.us-west-2.amazonaws.com/huggingface-pytorch-inference:2.0.0-transformers4.28.1-cpu-py310-ubuntu20.04

I am installing sam2 by adding this to requirement. text
git+https://github.com/facebookresearch/sam2.git

But I am getting this error

The NVIDIA driver on your system is too old (found version 11080). Please update your GPU driver by downloading and installing a new version from the URL: Download The Latest Official NVIDIA Drivers Alternatively, go to: https://pytorch.org to install a PyTorch version that has been compiled with your version of the CUDA driver.

There is not higher version cuda available in images list

How I can fix it

1 Like