如何检查Flux.jl是否正在使用图形处理器？

Question

我想检查Flux在我的计算机上使用的是GPU还是CPU。使用内置的Flux.jl函数可以做到这一点吗？

Answer 1

这似乎可以通过执行以下操作来实现：

julia> using Flux

julia> Flux.use_cuda[]
false

这似乎是可行的，因为我没有在我的Mac上运行CUDA，并且在幕后，Flux检查这个条件，然后将use_cuda标志设置为false。您可以在此处看到执行此操作的源代码：https://github.com/FluxML/Flux.jl/blob/de76e0853db8f0216addaaa4eacf6de39630d834/src/Flux.jl#L58，但它看起来类似于：

 use_cuda[] = CUDA.functional() # Can be overridden after load with `Flux.use_cuda[] = false`
 if CUDA.functional()
   if !CUDA.has_cudnn()
     @warn "CUDA.jl found cuda, but did not find libcudnn. Some functionality will not be available."
   end
 end