最近做计算机视觉学习,读项目的时候,需要配置PyTorch,遇到了一点问题,下面记录一下,主要分为如下几个步骤:
1.查看自己的GPU版本
(1)如果要查看显卡详细信息,我们首先根据自己电脑的版本,有选择的下载GPU-Z软件,双击打开,点击Yes进行安装。
到这步选择Next
选择安装路径,这里我安装到默认C盘不做更改,然后点击Install,就好了。
直到安装完成,close关掉,此时就安装成功了。
软件打开后,显示了gpu的详细信息
2.开始安装
从上图中可以看到我的是型号是11的。接下来去PyTorch官网,找对应的安装方法
点击GetStarted进入下一界面
根据你的版本进行复制安装信息,根据自己显卡的信息,进行选择CUDA,我这里是11,可以安装11以下的,这里python3.7我就选择3.7然后,复制到下面win+r–>cmd ,进入下面这个界面
输入下面的信息,就可以安装了
因为我安装好了,就不演示了,如有不正确的地方,欢迎交流讨论。