UniqAI/nvidia-container-toolkit
1
0
Fork 0
mirror of https://github.com/NVIDIA/nvidia-container-toolkit synced 2025-01-22 18:47:32 +00:00
Code Issues Actions 6 Packages Projects Releases Wiki Activity

Include HasNVML check in ResolveAutoMode

Browse Source 
Signed-off-by: Evan Lezar <elezar@nvidia.com>
This commit is contained in:
Evan Lezar 2022-05-11 14:52:59 +02:00
parent ad9ec1efae
commit ff86ecb2a5
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
internal/info/auto.go
View File

@ -35,7 +35,10 @@ func ResolveAutoMode(logger Logger, mode string) (rmode string) {
isTegra, reason := IsTegraSystem()
logger.Debugf("Is Tegra-based system? %v: %v", isTegra, reason)
if isTegra {
hasNVML, reason := HasNVML()
logger.Debugf("Has NVML? %v: %v", hasNVML, reason)
if isTegra && !hasNVML {
return "csv"
}