python 安装pytorch GPU

在Python中安装PyTorch并使其支持GPU通常需要以下几个步骤：

1. **检查环境**：
首先，确保你的计算机上已经安装了CUDA（NVIDIA计算平台的库），以及cuDNN（深度学习加速库）。你可以访问NVIDIA官网下载对应版本。

2. **安装Python和pip**：
确保你有最新版的Python（建议使用3.x版本）和pip工具。如果没有，可以从Python官方网站下载安装。

3. **安装Anaconda（可选）**：
Anaconda是一个流行的Python发行版，它包含了许多科学计算库，包括PyTorch。如果你对管理包有较高要求，可以考虑安装。

4. **通过Conda安装**：
使用命令行运行以下命令（如果你使用的是Anaconda或Miniconda）：

   conda create -n pytorchenv torch torchvision cudatoolkit=version_number anaconda

其中`version_number`是你要使用的CUDA版本号。例如，如果你的系统上有CUDA 11.0，那么写`cudatoolkit=11.0`。

5. **激活环境**：
创建完环境后，激活它：

   conda activate pytorchenv

6. **安装PyTorch**：
现在你可以使用`conda install`命令直接安装PyTorch：

   conda install pytorch torchvision cpuonly

如果你想指定GPU支持，去掉`cpuonly`。

7. **验证安装**：
安装完成后，导入PyTorch看看是否能识别到GPU：

   import torch
   print(torch.cuda.is_available())  # 输出True表示成功

8. **清理缓存**：
安装过程中可能会创建一些临时文件，可以用以下命令清理：

   conda clean --packages --tarballs --index-cache