python安装gpu-torch

1. 安装CUDA和cuDNN

要安装GPU版本的PyTorch，需要先安装CUDA和cuDNN。请根据你的GPU型号和操作系统版本到NVIDIA官网下载对应的CUDA和cuDNN安装包，并按照安装指南进行安装。

2. 创建虚拟环境

在安装PyTorch之前，我们建议使用conda或者虚拟环境来创建一个独立的Python环境，以避免与其他Python库发生冲突。

首先，安装conda（如果你还没有安装）：

wget https://repo.anaconda.com/archive/Anaconda3-2021.05-Linux-x86_64.sh
bash Anaconda3-2021.05-Linux-x86_64.sh
source ~/.bashrc

然后，创建一个新的conda环境：

conda create --name myenv python=3.8
conda activate myenv

3. 安装PyTorch

现在可以开始安装PyTorch了。打开终端，输入以下命令安装PyTorch：

conda install pytorch torchvision torchaudio cudatoolkit=11.1 -c pytorch -c nvidia

其中，`cudatoolkit=11.1` 是CUDA的版本号，需要根据你的实际安装情况进行修改。

安装完成后，可以在Python中测试是否成功安装：

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

如果输出`True`，说明已经成功安装了GPU版的PyTorch。