get some gpu information / published by https://github.com/dacr/code-examples-manager #2823b304-e6a7-47fb-80b2-4ca33fb3cc3d/5d5ff6a93ed7dd68bda667dff51a90b8980b8791

djl-gpu-info.sc

// summary : get some gpu information
// keywords : djl, machine-learning, gpu, @testable
// publish : gist
// authors : David Crosson
// license : Apache NON-AI License Version 2.0 (https://raw.githubusercontent.com/non-ai-licenses/non-ai-licenses/main/NON-AI-APACHE2)
// id : 2823b304-e6a7-47fb-80b2-4ca33fb3cc3d
// created-on : 2024-02-18T11:58:44+01:00
// managed-by : https://github.com/dacr/code-examples-manager
// run-with : scala-cli $file

// ---------------------
//> using scala "3.4.2"
//> using dep "org.slf4j:slf4j-api:2.0.13"
//> using dep "org.slf4j:slf4j-simple:2.0.13"
//> using dep "ai.djl:api:0.28.0"
//> using dep "ai.djl:basicdataset:0.28.0"
//> using dep "ai.djl:model-zoo:0.28.0"
//> using dep "ai.djl.pytorch:pytorch-engine:0.28.0"
//> using dep "ai.djl.pytorch:pytorch-model-zoo:0.28.0"

import ai.djl.Device
import ai.djl.util.cuda.CudaUtils
// ---------------------

val device = Device.gpu()
println(
  s"""
  cuda version : ${CudaUtils.getCudaVersion}
  cuda version string : ${CudaUtils.getCudaVersionString}
  gpu count : ${CudaUtils.getGpuCount}
  gpu memory : ${CudaUtils.getGpuMemory(device)}
  """
)