如何检查 Flux.jl 是否正在使用 GPU？

Question

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

Answer 1

这似乎是可行的方法是：

julia> using Flux

julia> Flux.use_cuda[]
false

这似乎可行，因为我没有 CUDA 在我的 Mac 上工作，并且在引擎盖下，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

如何检查 Flux.jl 是否正在使用 GPU？

How to check if Flux.jl is using the GPU?

julia

flux.jl