Pytorch_gpu 版本安装

准备工作一

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

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

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

点击链接,查询显卡是否在列表https://developer.nvidia.com/cuda-gpus
Pytorch_gpu 版本安装_第1张图片
3.如果支持CUDA,支持哪一版本的CUDA? 我的是支持CUDA10.2

查看方法一:

Pytorch_gpu 版本安装_第2张图片

查看方法二:

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

Pytorch_gpu 版本安装_第3张图片
Pytorch_gpu 版本安装_第4张图片

准备工作二

安装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

你可能感兴趣的:(安装)