确认本地 CUDA 版本为 11.1
直接使用 PyTorch 官方指定命令安装（推荐）
PyTorch 1.8.1+cu111 的官方安装命令已内置源地址，能避免手动指定版本的适配问题
# CPU+CUDA 11.1 组合（若仅需GPU版，可去掉torchvision和torchaudio）
pip3 install torch==1.8.1+cu111 torchvision==0.9.1+cu111 torchaudio==0.8.1 -f https://download.pytorch.org/whl/torch_stable.html