Pytorch_gpu 版本安装

准备工作一

1、查看自己的电脑是否支持GPU

怎样知道自己的电脑是否支持GPU？https://blog.csdn.net/YPP0229/article/details/106364757

2. 测试本机电脑是否支持CUDA的安装

点击链接，查询显卡是否在列表https://developer.nvidia.com/cuda-gpus

3.如果支持CUDA，支持哪一版本的CUDA？ 我的是支持CUDA10.2

查看方法一：

查看方法二：

查看适合自己电脑的CUDA版本：https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html#abstract：

准备工作二

安装VS2017

1. 下载：https://visualstudio.microsoft.com/zh-hans/downloads/
2. 安装参考教程：VS2017全部下载后再安装的路径在哪里？安装过程是什么？https://blog.csdn.net/libaineu2004/article/details/89554461

安装CUDA（全名，CUDA Toolkit）

1. 下载CUDA Toolkit：https://developer.nvidia.com/cuda-toolkit-archive

2. 安装CUDA Toolkit：windows10下安装GPU版pytorch简明教程：https://zhuanlan.zhihu.com/p/54350088

3. 测试是否安装成功：运行cmd，输入nvcc --version，即可查看版本号；

安装cudann

1. 下载cudann：https://developer.nvidia.com/cudnn
2. 将cudann解压到你想放的位置

最后，可以安装pytorch了

参考

1. windows10下安装GPU版pytorch简明教程：https://zhuanlan.zhihu.com/p/54350088
2. win10下pytorch-gpu安装以及CUDA详细安装过程https://blog.csdn.net/mind_programmonkey/article/details/99688839

测试GPU版本的Pytorch是否安装成功

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