Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Fails to load if nvidia mig driver used for gpu #4323

Open
1 task done
djjeffr opened this issue Feb 26, 2024 · 0 comments
Open
1 task done

Fails to load if nvidia mig driver used for gpu #4323

djjeffr opened this issue Feb 26, 2024 · 0 comments
Labels

Comments

@djjeffr
Copy link

djjeffr commented Feb 26, 2024

How to reproduce this bug?

in values yaml
muti2vec-clip:
...
enable_cuda: true
...
#added this line to make it work
cuda_visible_devices: 0
...
nvidia.com/mig-1g.5gb: 1

Also had to modiify file transformersInferenceDeployment.yaml to allow the line cuda_visible_devices
-helm chart 16.4.0

What is the expected behavior?

Was expecting it to work without adding cuda_visible_devices line, the same as it worked when using nvidia.com/gpu: 1

What is the actual behavior?

If cuda_visible_device is not there it fails to start with error cuda_visible_device must be there to use nvidia mig driver.

Supporting information

No response

Server Version

1.20.5

Code of Conduct

@djjeffr djjeffr added the bug label Feb 26, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
Projects
None yet
Development

No branches or pull requests

1 participant