Fix lookup of libcuda.so.RM_VERSION from ldcache

Signed-off-by: Evan Lezar <[email protected]>

Author: elezar

internal/lookup/cuda/cuda.go

@@ -17,6 +17,8 @@
 package cuda
 
 import (
+	"path/filepath"
+
 	"github.com/NVIDIA/nvidia-container-toolkit/internal/lookup"
 )
 
@@ -35,5 +37,19 @@ func New(libraries lookup.Locator) lookup.Locator {
 // Locate returns the path to the libcuda.so.RMVERSION file.
 // libcuda.so is prefixed to the specified pattern.
 func (l *cudaLocator) Locate(pattern string) ([]string, error) {
-	return l.Locator.Locate("libcuda.so" + pattern)
+	candidates, err := l.Locator.Locate("libcuda.so.1")
+	if err != nil {
+		return nil, err
+	}
+
+	var matched []string
+	for _, candidate := range candidates {
+		matches, err := filepath.Match("libcuda.so"+pattern, filepath.Base(candidate))
+		if err != nil || !matches {
+			continue
+		}
+		matched = append(matched, candidate)
+	}
+
+	return matched, nil
 }