UniqAI/nvidia-container-toolkit
1
0
Fork 0
mirror of https://github.com/NVIDIA/nvidia-container-toolkit synced 2024-11-25 21:39:10 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Add HasNVML function to check if NVML is supported

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

17
internal/info/tegra.go → internal/info/info.go
View File

@ -20,8 +20,25 @@ import (
"fmt" "fmt"
"os" "os"
"strings" "strings"
"github.com/NVIDIA/go-nvml/pkg/dl"
) )
// HasNVML returns true if NVML is detected on the sytems
func HasNVML() (bool, string) {
const (
nvmlLibraryName = "libnvidia-ml.so.1"
nvmlLibraryLoadFlags = dl.RTLD_LAZY
)
lib := dl.New(nvmlLibraryName, nvmlLibraryLoadFlags)
if err := lib.Open(); err != nil {
return false, fmt.Sprintf("could not load NVML: %v", err)
}
defer lib.Close()
return true, "found NVML library"
}
// IsTegraSystem returns true if the system is detected as a Tegra-based system // IsTegraSystem returns true if the system is detected as a Tegra-based system
func IsTegraSystem() (bool, string) { func IsTegraSystem() (bool, string) {
const tegraReleaseFile = "/etc/nv_tegra_release" const tegraReleaseFile = "/etc/nv_tegra_release"