Win10下配置Pytorch-GPU（CUDA10.1）

Win10下配置Pytorch-GPU（CUDA10.1）

一、 CUDA安装

官网点击此处，根据需要版本下载安装，这里我选择的是CUDA Toolkit 10.1 update2 (Aug 2019)版本

点击Download下载安装包即可，CUDA的安装就一直默认选项安装
在命令行窗口输入

nvcc -V

查验CUDA是否安装成功

可以看到CUDA的版本号

二、CUDNN下载

官网点击此处下载CUDNN，CUDNN的版本要与CUDA版本保持一致。这里我选择的版本如下：

下载完成解压缩，将CUDNN里面的bin、clude、lib文件直接复制到CUDA的安装目录，如下图为CUDA的安装位置，粘贴过来直接覆盖即可。

三、Pytorch-GPU安装

Pytorch官网点击此处，根据配置进行选择。这里我选择pip安装，由于网络问题和其他因素，pip下载很慢，这里做一些修改，从清华镜像下载，然后等待下载直至安装成功，若报错，多试几次就好。

pip install --default-timeout=1000 -i https://pypi.tuna.tsinghua.edu.cn/simple torch==1.5.1+cu101 torchvision==0.6.1+cu101 -f https://download.pytorch.org/whl/torch_stable.html

安装过程：

命令行验证pytorch是否安装成功

import torch
print(torch.__version__)

接下来验证能否使用gpu

print(torch.cuda.is_available())

返回True，安装成功！