CI: Update docker image

[akihironitta]

What does this PR do?

• Part of CI/CD: Add CUDA version to docker image tags Lightning-AI/pytorch-lightning#13831.
• Updates PyTorch version from 1.8 to 1.9 as we don't build 1.8 images in the PL repo anymore. (Let's see if the CI allows it...)

Before submitting

• Was this discussed/approved via a Github issue? (no need for typos and docs improvements)
• Did you read the contributor guideline, Pull Request section?
• Did you make sure your PR does only one thing, instead of bundling different changes together?
• [n/a] Did you make sure to update the documentation with your changes?
• [n/a] Did you write any new necessary tests? [not needed for typos/docs]
• [n/a] Did you verify new and existing tests pass locally with your changes?
• [n/a] If you made a notable change (that affects users), did you update the CHANGELOG?

PR review

• Is this pull request ready for review? (if not, please submit in draft mode)

Anyone in the community is free to review the PR once the tests have passed.
If we didn't discuss your PR in Github issues there's a high chance it will not be merged.

Did you have fun?

Make sure you had fun coding 🙃

[krshrimali]

Thanks! 🚀

[codecov]

Codecov Report

Merging #1411 (e7561a7) into master (2657315) will decrease coverage by 0.79%.
The diff coverage is n/a.

❗ Current head e7561a7 differs from pull request most recent head e59188d. Consider uploading reports for the commit e59188d to get more accurate results

@@            Coverage Diff             @@
##           master    #1411      +/-   ##
==========================================
- Coverage   91.79%   91.00%   -0.80%     
==========================================
  Files         286      286              
  Lines       12861    12861              
==========================================
- Hits        11806    11704     -102     
- Misses       1055     1157     +102     

Flag	Coverage Δ
unittests	91.00% <ø> (-0.80%)	⬇️

Flags with carried forward coverage won't be shown. Click here to find out more.

Impacted Files	Coverage Δ
flash/graph/classification/input.py	45.45% <0.00%> (-54.55%)	⬇️
flash/graph/classification/cli.py	38.46% <0.00%> (-53.85%)	⬇️
flash/graph/collate.py	50.00% <0.00%> (-50.00%)	⬇️
flash/graph/classification/model.py	42.62% <0.00%> (-49.19%)	⬇️
flash/graph/embedding/model.py	54.83% <0.00%> (-45.17%)	⬇️
flash/graph/backbones.py	64.28% <0.00%> (-35.72%)	⬇️
flash/graph/classification/input_transform.py	66.66% <0.00%> (-33.34%)	⬇️
flash/graph/classification/data.py	69.23% <0.00%> (-30.77%)	⬇️
flash/core/data/utilities/samples.py	87.50% <0.00%> (-12.50%)	⬇️
flash/text/question_answering/collate.py	95.29% <0.00%> (-4.71%)	⬇️

Help us with your feedback. Take ten seconds to tell us how you rate us. Have a feature suggestion? Share it here.

[krshrimali]

Please don't worry about the failures, the CI is failing, and I'm working on it: #1410. :)

[krshrimali]

A couple of examples failures, namely: Lightning-AI.lightning-flash (Examples) (domain text with GPUs 0) and Lightning-AI.lightning-flash (Examples) (domain video with GPUs 0,1) are flaky. The text example is failing with some connection error to HuggingFace.co and the video example is failing with CUDA Runtime Error, I'll be re-running these tests to ensure they pass, and then we can merge this. (Other failures are being fixed, should be good very soon)

[akihironitta]

@krshrimali Thank you, Kush, for having a look at this! I'll re-re-run these three failing jobs and see if they pass this time :)

Lightning-AI.lightning-flash (Examples) Required
Lightning-AI.lightning-flash (Examples) (domain video with GPUs 0,1)
Lightning-AI.lightning-flash (Special)