How to specify GPU id in nuclio functions?

[yuanyuangoo]

I am using sam to help the labelling, which use a GPU. What should I do to specify a certain GPU to use?

[315386775]

Good job. with sam model the cvat may get more stars and attention.

[bsekachev]

Probably it can be configured via configuration .yml file.

[315386775]

@bsekachev are there any plans to integrate SAM into CVAT.

[yuanyuangoo]

@bsekachev are there any plans to integrate SAM into CVAT.

I am using it.

[bsekachev]

@315386775

It is integrated.

#6008
#6019

https://www.cvat.ai/post/facebook-segment-anything-model-in-cvat

[yuanyuangoo]

Get Outlook for iOS<https://aka.ms/o0ukef>

…

________________________________ From: MarkAI ***@***.***> Sent: Thursday, May 11, 2023 1:45:52 PM To: opencv/cvat ***@***.***> Cc: Alan Shangguan ***@***.***>; Author ***@***.***> Subject: Re: [opencv/cvat] How to specify gpu id for serverless SAM fuction? (Issue #6104) @bsekachev<https://github.com/bsekachev> are there any plans to integrate SAM into CVAT. — Reply to this email directly, view it on GitHub<#6104 (comment)>, or unsubscribe<https://github.com/notifications/unsubscribe-auth/AC5SJ57LDLZSOCP4TOQAIBDXFRAFBANCNFSM6AAAAAAXVFTANM>. You are receiving this because you authored the thread.Message ID: ***@***.***>

[twmht]

@bsekachev

Can we integrate our own model with cvat? and of tutorial?

[silverlining21]

Change the follow line into
https://github.com/opencv/cvat/blob/f52f0f3e0ea09b72bc3efdead9780ca2fb2a858c/serverless/pytorch/facebookresearch/sam/nuclio/function-gpu.yaml#L36

value: NVIDIA_VISIBLE_DEVICES=2

Make sure you specify your GPU id during image building. Make sure your env works well for the gpu, such as nv driver nv-docker all sounds work well.
enjoy！