CUDA11.4安装gpu版pytorch

以下是在Windows系统上安装GPU版PyTorch的步骤：

1. 安装CUDA11.4

在CUDA官网下载CUDA11.4的安装程序，并按照提示完成安装。安装完成后，将CUDA的bin和lib路径添加到环境变量中。

2. 安装cuDNN

在NVIDIA官网下载cuDNN的安装程序，并按照提示完成安装。安装完成后，将cuDNN的bin和lib路径添加到环境变量中。

3. 安装Anaconda

在Anaconda官网下载Anaconda的安装程序，并按照提示完成安装。

4. 创建虚拟环境

打开Anaconda Prompt，执行以下命令创建一个名为pytorch的虚拟环境：

conda create -n pytorch python=3.8

5. 激活虚拟环境

执行以下命令激活pytorch虚拟环境：

conda activate pytorch

6. 安装PyTorch

执行以下命令安装PyTorch：

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

7. 验证安装

执行以下命令，验证PyTorch是否安装成功：

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

如果输出结果为True，则说明PyTorch安装成功，并且GPU也已经被正确识别。