Work around NVML issue on Jetson Orin. (#2620)

JuliaGPU · Jan 15, 2025 · fe0419a · fe0419a
1 parent a4a9166
commit fe0419a
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/lib/nvml/NVML.jl b/lib/nvml/NVML.jl
@@ -30,6 +30,12 @@ end
 
 function has_nvml()
     @memoize begin
+        if CUDA.is_tegra()
+            # XXX: even though Orin supports NVML, we don't know how to
+            #      look up the device (CUDA.jl#2580)
+            return false
+        end
+
         if Libdl.dlopen(libnvml(); throw_error=false) === nothing
             return false
         end

diff --git a/src/utilities.jl b/src/utilities.jl
@@ -139,7 +139,6 @@ function versioninfo(io::IO=stdout)
             try
                 query_nvml()
             catch err
-                @show err
                 if !isa(err, NVML.NVMLError) ||
                    !in(err.code, [NVML.ERROR_NOT_SUPPORTED, NVML.ERROR_NO_PERMISSION])
                     rethrow()