pytorch，才会采用清华源安装。12345673、安装完成后，测试torch和CUD

PyTorch是一个开源的深度学习框架，由Facebook的AI研究部门开发。它提供了一个动态计算图库，非常适合研究实验以及快速原型设计。当需要从清华大学源而非默认的官方源安装PyTorch时，通常是因为特定版本在国内访问更方便，或者是为了绕过可能存在的网络限制。

1. 安装步骤（假设你是Windows用户，使用conda）：
- 打开命令行终端，输入`conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch`
- 更新配置后，可以安装PyTorch：`conda install pytorch torchvision cudatoolkit=10.2 -c pytorch`
或者根据你的CUDA版本选择合适的 cudatoolkit 版本。

2. 完成安装后，你可以通过以下代码测试是否成功安装了torch和CUDA支持：

import torch
# 检查是否有GPU可用
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
print(f"当前设备: {device}")

# 创建一个张量并转移到GPU
x = torch.rand(1, 2, device=device)
print(x)

# 输出一些基本信息，如PyTorch版本等
print(torch.__version__)