Skip to content

Conversation

@sonic2kk
Copy link
Owner

@sonic2kk sonic2kk commented Oct 23, 2023

Should fix #953.

When USEPRIMERUN is on, we export several environment variables. But if the nvidia_icd.json is not set in the current environment's VK_ICD_FILENAMES, it will not be able to use the Nvidia GPU. For cases likee this, we should explicitly export nvidia_icd.json.

The fix for this was given by OP in #953 (they informed which environment variable to use), this PR just applies it in the appropriate spot in the code.

This path is not likely to change, and if other custom ones are necessary, a user can manually set these in their STL env variable config file.

Putting this up as a branch in case this fix is not sufficient, and we have to make other changes.

@sonic2kk sonic2kk changed the title Primerun: Ensure nvidiaa_icd.json is used Primerun: Ensure nvidia_icd.json is used Oct 23, 2023
@sonic2kk
Copy link
Owner Author

OP confirmed it to work, shellcheck is green. Ready to merge.

@sonic2kk sonic2kk merged commit 87bb53a into master Oct 23, 2023
@sonic2kk sonic2kk deleted the primerun-change branch October 23, 2023 17:30
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

primerun setting doesn't work

2 participants