pytorch 核心显卡 pytorch显卡要求

准备阶段：

     一、Cuda的安装及其配置

              （1）通过查看本机的显卡配置，进行选择Cuda的类型。在本机查看显卡的操作步骤为：计算机->管理->设备管理器->显示适配器。（我感觉，其实就是预估你计算机的GPU计算机能力，如果列表中没有你的显卡类型，在网上查一哈那个显卡的好坏，可以选择比你显卡低的，如果选择比你的高的，你的显卡可能会带不动。这个是个人的一个想法，没有得到检验，哈哈哈哈）

查看计算机的计算机能力请点击这里

你将会看见：

点击第四个

之后就可以根据你电脑的计算能力选择相应的版本进行安装了。点击这里进行下载

接下来的安装下一步下一步即可，但需要注意的是，如果本机中没有VS，则将自定义安装中的vs选项去掉，由于我是安装后写的博客，没有相应的截图啦，请移步到文首的博客中查看，我写这个博客的原因就是为了记录一下自己学习的过程。

安装完成过后，进行系统环境变量的配置。

然后，进入cmd命令行，输入 nvcc -V（注意大小写）

恭喜你，Cuda安装完成

       二、CUDNN的安装

            1、请点击这里进行下载，根据第一步选择的Cuda版本来选择CUDNN版本，借用一下开头提到的博客的图片

下载后，将压缩包里的bin、include、lib文件直接拷到你安装的Cuda安装目录下，覆盖掉原来的文件即可。

      三、pytorch的安装

点击这里选择相应安装方式的命令，如下图

利用上面的命令安装失败的话，就直接到镜像下进行下载，参考博主的命令，镜像地址可以变为你知道的下载快的镜像地址。

pip3 install -i https://pypi.tuna.tsinghua.edu.cn/simple torch==1.2.0 torchvisinotallow==0.4.0 -f https://download.pytorch.org/whl/torch_stable.html

 

然后，导入 torch进行验证，import torch没有报错的话，恭喜你，pytorch安装成功！