Better way to resolve DLL path

mxpv · Feb 26, 2018 · f7df3fb · f7df3fb
1 parent dd19dbd
commit f7df3fb
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/bindings.go b/bindings.go
@@ -2,6 +2,7 @@ package nvml
 
 import (
 	"C"
+	"os"
 	"syscall"
 	"unsafe"
 
@@ -158,7 +159,7 @@ func (a API) ErrorString(result uintptr) string {
 // New creates nvml.dll wrapper
 func New(path string) (*API, error) {
 	if path == "" {
-		path = "C:\\Program Files\\NVIDIA Corporation\\NVSMI\\nvml.dll"
+		path = os.ExpandEnv("$ProgramW6432\\NVIDIA Corporation\\NVSMI\\nvml.dll")
 	}
 
 	dll, err := syscall.LoadDLL(path)