Available devices:
  CUDA0: NVIDIA RTX Spark N1X (6144-core Blackwell RTX GPU) (22950 MiB, 22551 MiB free)
