r00bi
(shebr)
2
There are 8 16G GPUS:
trainer = pl.Trainer(
accelerator=“auto”,
auto_select_gpus=True,
callbacks=callbacks,
devices=_get_devices(),
max_epochs=config[“max_epochs”],
strategy=_get_strategy(),
)
# Train the model.
trainer.fit(
model, train_loader.train_dataloader(), val_loader.val_dataloader()
)
I learned that why the gpus will be invisible please read this:
Thanks!