准备工作一
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
- 下载:https://visualstudio.microsoft.com/zh-hans/downloads/
- 安装参考教程:VS2017全部下载后再安装的路径在哪里?安装过程是什么?https://blog.csdn.net/libaineu2004/article/details/89554461
安装CUDA(全名,CUDA Toolkit)
-
下载CUDA Toolkit:https://developer.nvidia.com/cuda-toolkit-archive
-
安装CUDA Toolkit:windows10下安装GPU版pytorch简明教程:https://zhuanlan.zhihu.com/p/54350088
-
测试是否安装成功:运行cmd,输入
nvcc --version
,即可查看版本号;
安装cudann
- 下载cudann:https://developer.nvidia.com/cudnn
- 将cudann解压到你想放的位置
最后,可以安装pytorch了
参考
- windows10下安装GPU版pytorch简明教程:https://zhuanlan.zhihu.com/p/54350088
- win10下pytorch-gpu安装以及CUDA详细安装过程https://blog.csdn.net/mind_programmonkey/article/details/99688839
测试GPU版本的Pytorch是否安装成功
import torch
print(torch.cuda.is_available())