GPU版pytorch安装

0 设备环境：

Windows10 64bit

GPU: GeForce GTX 1050 Ti

Python3.7 64bit

1 下载安装 CUDA以及cuDNN

1.1 查看电脑显卡对应的CUDA驱动版本

1.2 下载安装CUDA

下载地址：https://developer.nvidia.com/cuda-downloads?target_os=Windows&target_arch=x86_64&target_version=10&target_type=exelocal

安装CUDA时选择自定义安装，并将Visual Studio Integration一项去掉打勾，否则有可能安装出错

检验安装是否安装正确：win+R→cmd→nvcc -V

1.3 下载cuDNN

下载地址：https://developer.nvidia.com/rdp/cudnn-archive

将cuDNN解压后的文件夹中bin、include和lib三个文件夹复制粘贴到CUDA安装路径中这三个文件夹的父文件夹下面，Windows中默认安装路径为：C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0

1.4 CUDA和cuDNN的作用

CUDA是NVIDIA推出的用于自家GPU的并行计算框架，也就是说CUDA只能在NVIDIA的GPU上运行，而且只有当要解决的计算问题是可以大量并行计算的时候才能发挥CUDA的作用。

cuDNN是一个常见的神经网络层加速库文件，其能够很大程度的把加载到显卡上的网络层数据进行优化计算，cuDNN需要在有cuda的基础上进行。

2 下载安装GPU版的Pytorch

下载地址：https://pytorch.org/get-started/locally/

选择对应版本信息，利用Conda或Pip进行安装，如外网下载太慢，可配置镜像源进行安装。

conda镜像源配置：https://blog.csdn.net/weixin_34019929/article/details/90985190

pip镜像源配置：https://blog.csdn.net/sunyllove/article/details/81627281

验证pytorch的gpu是否可用，返回结果为True则表示GPU可用。