Teste Torchx
Mix.install(
[
{:torchx, "~> 0.9.2"},
{:bumblebee, "~> 0.6.0"},
{:nx, "~> 0.9.2"}
],
config: [
nx: [default_backend: {Torchx.Backend, device: :vulkan}]
]
)Section
Torchx.device_count(:vulkan)Nx.tensor([1,1,1])
