Task: Configure PyTorch with CUDA Support for NVIDIA A100 GPU

Task: Configure PyTorch with CUDA Support for NVIDIA A100 GPU: a task in Terminal-Lego-15k (Harbor dataset). Write a Python diagnostic script and a shell setup script that correctly identifies and resolves PyTorch + CUDA compatibility issues for NVIDIA A100 GPUs (compute capability sm 80).