安装支持CUDA的PyTorch

1. ​​卸载当前PyTorch​​：

   pip uninstall torch

2. ​​安装CUDA兼容版本​​：

   # 示例：CUDA 11.8版本（根据您显卡的CUDA版本选择） pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118

3. ​​验证安装​​：

   import torch print(torch.__version__) # 查看版本 print(torch.cuda.is_available()) # 应返回True print(torch.version.cuda) # 显示CUDA版本