看过我之前那篇【Pytorch安装 CPU版本】的朋友应该知道,Pytorch安装步骤很复杂。
但是其实弄清楚整个逻辑会简单很多,本篇为PytorchGPU安装。如果有朋友跟着我之
前的帖子安装过的,想再安装一个GPU的,可以跟着我一起安装一下。我会把我遇到
的坑以及经验都给大家分享一下。
Pytorch GPU安装整体分为几个步骤:
1.之前是否安装过CPU版本的Pytorch
2.第一次安装。
步骤1:检查电脑是否支持NVIDIA。
步骤2:如何安装CUDA。
步骤3:如何安装cudNN。
步骤4:如何安装Pytorch。
步骤5:检查是否安装成功。
接下来我们就一步一步去讲解。
步骤1:查电脑是否支持NVIDIA。
如果是看过我之前那篇帖子的友友们,应该知道哈,我这里就直接说一种方法,能够
往下面走就行。
①cmd命令行[nvidia-smi],可以看到版本,说明有GPU。
②