老饼讲解-深度学习 机器学习 神经网络 深度学习
pytorch入门
1.pytorch学前准备
2.pytorch的基础操作
3.pytorch的梯度计算
4.pytorch的数据
5.pytorch-小试牛刀

【查看】查看pytorch相关信息

作者 : 老饼 发表日期 : 2023-07-28 10:52:28 更新日期 : 2024-01-18 12:35:01
本站原创文章,转载请说明来自《老饼讲解-深度学习》www.bbbdata.com





    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 







联系老饼