在下载pytorch的时候如何查看GPU型号以及pytorch的安装

问题分析

在下载pytorch的环境的时候我一直搞不明白在pytorch官网当中Compute Platform 这一栏的要求是什么,可能不是所有人都有这方面的问题,在这里进行一个问题的总结
首先进入到pytorch下载的官网

pytorch.org

在选取这个内容的时候,对如何查找这部分内容产生了疑惑

问题的解决

这一部分要求我们要查看GPU的型号,那么如何查看GPU的型号呢?

查看GPU的型号

win10
右键点击桌面的此电脑,选择管理，成功打开计算机管理界面，在侧边栏选择系统工具x里的设备管理器，然后在右侧选择显示适配器

如下得到电脑的显卡显示,可以根据自己的显卡能力选择是否使用GPU下载

还是不了解选哪一个最合适的,就要用到另一种方法了

查看显卡型号

桌面右键,选择NVIDIA的控制面板

鼠标放到帮助然后选择系统信息

点击组件,然后就可以查看CUDA了

安装pytorch

根据我的CUDA的版本,我选择了11.6,然后复制下载所给到的指令

打开Anaconda Prompt粘贴指令(我是下载到了D盘,如果后续会因为下载到D盘出现一些问题我也会第一时间进行更新)

使用pytorch

因为下载pytorch是因为在学习李沐老师的动手深度学习的过程当中是需要,pytorch 的环境的,我们尝试在pytorch下的环境进行运行
注在这里遇到了另一个问题就是李沐老师有一个d2l的一个包,需要另行操作,大家可以搜索一下,或者之后我之后再整理整理
下面以查看pytorch的版本作为例子

import torch
torch.__version__

运行之后得到如下

这就证明pytorch成功下载

小结

在遇到环境等问题的时候大家不要慌,就要学会先上网自行查看,要学会自动检索.