LightningDataset currently only supports 'SingleDevicePlugin' and 'DDPSpawnPlugin'

[tuanmp]

Hello,

I am trying to run the GIN+pytorch_lightning example under . I have torch_geometric 2.0.4 and PL 1.6.4 installed. If I run the example as is, I got this error

  File "gin.py", line 90, in <module>
    main()
  File "gin.py", line 85, in main
    trainer.fit(model, datamodule)
  File "/global/cfs/cdirs/m3443/usr/pmtuan/conda/exatrkx_hsf/lib/python3.8/site-packages/pytorch_lightning/trainer/trainer.py", line 770, in fit
    self._call_and_handle_interrupt(
  File "/global/cfs/cdirs/m3443/usr/pmtuan/conda/exatrkx_hsf/lib/python3.8/site-packages/pytorch_lightning/trainer/trainer.py", line 723, in _call_and_handle_interrupt
    return trainer_fn(*args, **kwargs)
  File "/global/cfs/cdirs/m3443/usr/pmtuan/conda/exatrkx_hsf/lib/python3.8/site-packages/pytorch_lightning/trainer/trainer.py", line 811, in _fit_impl
    results = self._run(model, ckpt_path=self.ckpt_path)
  File "/global/cfs/cdirs/m3443/usr/pmtuan/conda/exatrkx_hsf/lib/python3.8/site-packages/pytorch_lightning/trainer/trainer.py", line 1164, in _run
    self._data_connector.prepare_data()
  File "/global/cfs/cdirs/m3443/usr/pmtuan/conda/exatrkx_hsf/lib/python3.8/site-packages/pytorch_lightning/trainer/connectors/data_connector.py", line 124, in prepare_data
    self.trainer.datamodule.prepare_data()
  File "/global/cfs/cdirs/m3443/usr/pmtuan/conda/exatrkx_hsf/lib/python3.8/site-packages/torch_geometric/data/lightning_datamodule.py", line 55, in prepare_data
    raise NotImplementedError(
NotImplementedError: 'LightningDataset' currently only supports 'SingleDevicePlugin' and 'DDPSpawnPlugin' training type plugins of 'pytorch_lightning' (got 'DDPSpawnStrategy')
pmtuan@nid003725:/global/cfs/cdirs/m3443/usr/pmtuan/test_pyg_pl> python gin.py ```

When I tried to disable the PL trainer's DDP strategy, by 

`trainer = pl.Trainer(
        accelerator='gpu', devices=[0],
                         callbacks=[checkpoint], max_epochs=50, log_every_n_steps=5)`

I got 

```Traceback (most recent call last):
  File "gin.py", line 90, in <module>
    main()
  File "gin.py", line 85, in main
    trainer.fit(model, datamodule)
  File "/global/cfs/cdirs/m3443/usr/pmtuan/conda/exatrkx_hsf/lib/python3.8/site-packages/pytorch_lightning/trainer/trainer.py", line 770, in fit
    self._call_and_handle_interrupt(
  File "/global/cfs/cdirs/m3443/usr/pmtuan/conda/exatrkx_hsf/lib/python3.8/site-packages/pytorch_lightning/trainer/trainer.py", line 723, in _call_and_handle_interrupt
    return trainer_fn(*args, **kwargs)
  File "/global/cfs/cdirs/m3443/usr/pmtuan/conda/exatrkx_hsf/lib/python3.8/site-packages/pytorch_lightning/trainer/trainer.py", line 811, in _fit_impl
    results = self._run(model, ckpt_path=self.ckpt_path)
  File "/global/cfs/cdirs/m3443/usr/pmtuan/conda/exatrkx_hsf/lib/python3.8/site-packages/pytorch_lightning/trainer/trainer.py", line 1164, in _run
    self._data_connector.prepare_data()
  File "/global/cfs/cdirs/m3443/usr/pmtuan/conda/exatrkx_hsf/lib/python3.8/site-packages/pytorch_lightning/trainer/connectors/data_connector.py", line 124, in prepare_data
    self.trainer.datamodule.prepare_data()
  File "/global/cfs/cdirs/m3443/usr/pmtuan/conda/exatrkx_hsf/lib/python3.8/site-packages/torch_geometric/data/lightning_datamodule.py", line 55, in prepare_data
    raise NotImplementedError(
NotImplementedError: 'LightningDataset' currently only supports 'SingleDevicePlugin' and 'DDPSpawnPlugin' training type plugins of 'pytorch_lightning' (got 'SingleDeviceStrategy')

Any suggestion?

[rusty1s]

PyG 2.0.4 requires PL < 1.6. you can either downgrade PL or upgrade PyG to latest master. I will craft a new release soon.

[tuanmp]

Thanks for the quick answer. Will the new release be compatible with PL > 1.6?

[rusty1s]

Yes, it is.