【查看】查看pytorch相关信息

本站原创文章，转载请说明来自《老饼讲解-深度学习》www.bbbdata.com

在安装pytorch后，往往想查看GPU信息、查看CUDA版本、查看cudn版本，

本文讲解如何在pytorch中查看GPU、CUDA、cudn等相关信息

    01. 通过pytorch看一看GPU相关信息    

本节简单地利用pytorch来看一看GPU的信息

        通过pytorch看看显卡       

 下面，先通过pytorch看看显卡是怎么样的
  查看显卡的相关信息的代码如下

import torch
import math 
import numpy as np
from torch import nn
is_available = torch.cuda.is_available()              # 显卡个数
          
device_count = torch.cuda.device_count()              # 显卡个数
current_device = torch.cuda.current_device()          # 查看当前使用的显卡(序号)
device_name=torch.cuda.get_device_name()              # 查看显卡的名称
device_capability= torch.cuda.get_device_capability() # 支持CUDA的最高最低版本

print('is_available:',is_available)
print('device_count:',device_count)
print('current_device:',current_device)
print('device_name:',device_name)
print('device_capability:',device_capability)

运行结果显示如下

    02. 通过pytorch、CUDA、cudnn的版本    

本节简单地利用pytorch来看一看pytorch、CUDA、cudn的版本

     看看pytorch、CUDA、cudnn的版本信息    

看看安装的pytorch相关的软件的版本信息，代码如下

import torch
print('Pytorch版本:',torch.__version__)
print('CUDA版本:',torch.version.cuda)
print('cudnn版本:',torch.backends.cudnn.version())

运行结果显示如下
 

 End