深度学习之环境配置--安装和配置PyTorch(GPU版本)

在此之前我们已经成功安装好了PyTorch的CUP版本以及anaconda的两个环境(一个虚拟环境，一个base)，然后我们就可以在虚拟环境中安装GPU版本了。

1.安装CUDA

2.安装CUDNN

3.安装PyTorch的GUP版本

打开PyTorch官网进行如下选择：

4.检测环境

安装完成后，我们输入

python
import torch
torch.cuda.is_available()

如果返回值为True,那么恭喜你GPU环境安装成功.