Pytorch教程

【查看】查看pytorch相关信息

作者 : 老饼 发表日期 : 2023-07-28 10:52:28 更新日期 : 2024-04-22 01:18:52
本站原创文章,转载请说明来自《老饼讲解-深度学习》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 







联系老饼