Update CIFAR10 tutorial device selection for CUDA, MPS, and CPU#3826

amit-chaubey wants to merge 2 commits intopytorch:mainpytorch/tutorials:mainfrom

amit-chaubey:blitz-cifar10-device-optionsamit-chaubey/tutorials-pytorch:blitz-cifar10-device-optionsCopy head branch name to clipboard