如何安装社区版的torch

安装社区版的PyTorch（包括其依赖如CUDA）通常需要几个步骤，假设你在Windows、macOS或Linux系统上操作：

1. **Python环境**:
确保已安装Python，可以从Python官网下载并安装最新稳定版本。

2. **包管理器**:
对于大部分情况，你可以使用pip（Python包管理工具）来安装PyTorch。如果你还没有安装pip，可以先安装它。

- **对于Windows**: 打开命令提示符（cmd），输入 `python get-pip.py` 安装pip。
- **对于macOS/Linux**: 打开终端，运行 `curl https://bootstrap.pypa.io/get-pip.py | python`。

3. **CUDA支持**:
PyTorch需要NVIDIA CUDA支持。首先检查你的系统是否支持CUDA，访问NVIDIA官网找到适合的CUDA版本。然后下载cuDNN库（针对GPU加速的支持库）。

4. **安装PyTorch**:
使用pip安装社区版PyTorch，例如安装预训练模型（不含GPU）:

   pip install torch torchvision

如果你想利用GPU加速，安装时加上 `--cuda` 参数，比如:

   pip install torch torchvision --extra-index-url https://download.pytorch.org/whl/cu102/torch_stable.html

将`cu102`替换为你系统的CUDA版本号。

5. **验证安装**:
安装完成后，打开Python交互模式，导入torch模块来确认安装成功，并查看GPU可用性:

   import torch
   print(torch.cuda.is_available())