linux安装pytorch gpu

1. 安装CUDA和cuDNN

首先，需要安装CUDA和cuDNN。CUDA是NVIDIA提供的用于GPU加速计算的平台，而cuDNN是NVIDIA提供的用于深度学习的GPU加速库。在安装之前，需要确认自己的GPU型号和CUDA版本是否匹配。可以在NVIDIA官网上查找相应的版本。

2. 安装Anaconda

Anaconda是一个Python的集成环境，可以方便地管理Python包和环境。可以在Anaconda官网上下载对应操作系统的安装包，安装完成后，可以使用conda命令来管理Python环境和包。

3. 创建虚拟环境

在安装pytorch之前，需要创建一个虚拟环境，以避免与其他Python包产生冲突。可以使用conda命令来创建虚拟环境：

conda create -n pytorch_env python=3.7

其中，pytorch_env是虚拟环境的名称，python=3.7表示使用Python 3.7版本。

4. 激活虚拟环境

创建虚拟环境后，需要激活虚拟环境，以便在该环境下安装和使用pytorch。可以使用以下命令来激活虚拟环境：

conda activate pytorch_env

其中，pytorch_env是虚拟环境的名称。

5. 安装pytorch

可以使用conda命令来安装pytorch：

conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch

其中，cudatoolkit=10.2表示使用CUDA 10.2版本，可以根据自己的GPU型号和CUDA版本进行选择。

6. 测试pytorch

安装完成后，可以使用以下代码来测试pytorch是否正常工作：

import torch

print(torch.cuda.is_available())

如果输出True，则表示pytorch已经成功安装并可以使用GPU加速。