Support Python 3.10

[akihironitta]

🚀 Feature

We should support Python 3.10 as Python 3.10 was released at the beginning of this month.

Required dependencies supporting Python 3.10

• [x] numpy>=1.21.0 https://numpy.org/news/#numpy-1210-release
• [x] torch>=1.11.0 https://github.com/pytorch/pytorch/issues/66424
• [x] tqdm
• [x] PyYAML
• [x] fsspec
• [x] tensorboard https://github.com/tensorflow/tensorboard/releases/tag/2.8.0
• [x] torchmetrics since its dependency torch>=1.11 supports Python 3.10
• [x] pyDeprecate https://github.com/Borda/pyDeprecate/issues/18
• [x] packaging
• [x] typing-extensions

The above checklist is mainly based on PyPI badge on pypi.org

---

If you enjoy Lightning, check out our other projects! ⚡

• Metrics: Machine learning metrics for distributed, scalable PyTorch applications.

• Flash: The fastest way to get a Lightning baseline! A collection of tasks for fast prototyping, baselining, finetuning and solving problems with deep learning

• Bolts: Pretrained SOTA Deep Learning models, callbacks and more for research and production with PyTorch Lightning and PyTorch

• Lightning Transformers: Flexible interface for high performance research using SOTA Transformers leveraging Pytorch Lightning, Transformers, and Hydra.

cc @borda @carmocca @akihironitta @tchaton @rohitgr7

[Borda]

let's try but what I tried last time still does not have the full support of all our required packages (dependencies)

[stale[bot]]

This issue has been automatically marked as stale because it hasn't had any recent activity. This issue will be closed in 7 days if no further activity occurs. Thank you for your contributions, Pytorch Lightning Team!

[Borda]

Let's try now when PT 1.11 is out... 🐰

[carmocca]

@akihironitta is this done?

[akihironitta]

@carmocca Will start working on it :)

[akihironitta]

TODO:

• [x] Enable Python 3.10 in "full" CPU testing: #13829
• [ ] Enable Python 3.10 in "conda" CPU testing <- (kind of) BLOCKED by #14059.
• [x] Enable Python 3.10 in GPU testing <- blocked by #13831. Also, no PyPI distribution of bagua with Python 3.10.

[carmocca]

I believe this can be closed now @akihironitta

[carmocca]

One follow-up improvement would be to use 3.10 in the mypy job cc @otaj

[function2-llx]

Hello, want to ask here if Python 3.10 is supported now? The badge seems not to include 3.10.

[carmocca]

It is! Would you like to open a PR updating the badge?

[function2-llx]

I'd love to!