windows下搭建pytorch框架

参考网址：https://zhuanlan.zhihu.com/p/54350088

1. 安装nvidia驱动

2. 安装cuda

3. 安装cudnn

4. 安装pytorch

主流安装方式有两种，分别是conda和pip。
建议使用conda方式，因为其依托强大的Anaconda集成开发环境，对新手更加友好。
1）安装Anaconda
2）通过以下命令，创建一个独立的运行环境，方便开发过程中遇到的多版本并存
powershell conda create -n pytorch_gpu pip python=3.6
3）通过以下命令，进入创建的环境
powershell conda activate pytorch_gpu
重点：通过以下命令，安装pytorch，
powershell conda install pytorch torchvision cuda100 -c pytorch

5.验证

1）通过以下命令，验证pytorch安装成功
powershell import torch x = torch.rand(5,3) print(x)

输出以下内容，安装成功：

	tensor([[0.3380, 0.3845, 0.3217],
    [0.8337, 0.9050, 0.2650],
    [0.2979, 0.7141, 0.9069],
    [0.1449, 0.1132, 0.1375],
    [0.4675, 0.3947, 0.1426]])

2）通过以下命令，验证gpu和cuda配置成功：
powershell torch.cuda.is_available()
如果输出“True”，则说明配置成功！