pytorch_lightning.utilities.exceptions.MisconfigurationException: The provided lr scheduler `CosineAnnealingLR` doesn't follow PyTorch's LRScheduler API. You should override the `LightningModule.lr_scheduler_step` hook with your own logic if you are using a custom LR scheduler. about mobiler2l HOT 2 CLOSED

I can roughly confirm that there is a compatibility issue between versions pytorch, cuda, and pytorch-lighting.

So how should I configure the ngp_pl environment on my device, which is cuda=11.8?

from mobiler2l.

Hi, ngp_pl environment is bit tricky to setup as its dependecies such as torch-scatter and apex needs a specific cuda version. That's why we recommend to use a sepertate conda environment with cuda=11.6 for ngp_pl. I remember apex will throw some building errors if using torch=2.0 + cuda=11.7 back then. That being said, it would be easier to use the recommended configurations to setup ngp_pl instead of modifying the codebase.

from mobiler2l.