docker: simplify the docker name with CUDA#21001

lantiga merged 2 commits intomasterLightning-AI/pytorch-lightning:masterfrom

docker/cudaLightning-AI/pytorch-lightning:docker/cudaCopy head branch name to clipboard