pytorch 检测gpu pytorch查看显卡

文章目录

• 查看NVIDIA驱动

• 查看信息
• 下载

• Pytorch 安装（Anaconda）

• 查看环境
• 创建环境
• 下载pytorch库
• 检查安装是否成功

查看NVIDIA驱动

显卡驱动程序就是用来驱动显卡的程序，它是硬件所对应的软件。驱动程序即添加到操作系统中的一小块代码，其中包含有关硬件设备的信息。正常有显卡的电脑都是有驱动程序的，但是有的时候驱动可能版本比较低，支持的cuda版本也是比较低。

查看信息

第一步查看电脑显卡是啥；

方法一：此电脑 -->高级系统设置–>显示适配器

方法二（需要有显卡驱动）：

在右下角找到这个小图标打开，或者右键桌面找到控制面板

下载

了解情况后去英伟达官网上去找相对应的显卡驱动更新或者下载。官网：

https://www.nvidia.cn/Download/index.aspx?lang=cn 找到自己的然后搜索，在下载。

下载完后，通过win+r ,打开cmd。
输入：

nivdia-smi

若能弹出一下界面，就ok了。

Pytorch 安装（Anaconda）

更多conda使用

查看环境

conda env list

最开始我们只有base一个环境，后续可以自己添加。这次我们再添加一个torch_env环境。

创建环境

conda create -n torch_env python=3.8

下载完后的状态

这个时候我们再检查环境

conda env list

舒服了，继续下一步。打开刚刚创建的环境

conda activate torch_env

下载pytorch库

打开https://pytorch.org/，开头我们通过驱动检测到我们的显卡是什么，最高支持cuda版本是什么。根据信息选择我们相应的版本，这里我选的10.2

复制贴贴到终端（每个人不一样的，还是自己去官网选择）

conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch

检查安装是否成功

import torch
print(torch.cuda.is_available())
print(torch.backends.cudnn.is_available())
print(torch.cuda_version)
print(torch.backends.cudnn.version())

这样就好了