python代码查看torch啊 cuda啊显卡啊这些玩意儿

最新推荐文章于 2024-05-19 23:31:32 发布

饿了就干饭

最新推荐文章于 2024-05-19 23:31:32 发布

阅读量457

点赞数 8

分类专栏： Pytorch 文章标签： python 深度学习 pytorch

本文链接：https://blog.csdn.net/qq_45893319/article/details/135821548

版权

Pytorch 专栏收录该内容

14 篇文章 0 订阅

订阅专栏

代码

import torch
print('CUDA版本:',torch.version.cuda)
print('Pytorch版本:',torch.__version__)
print('显卡是否可用:','可用' if(torch.cuda.is_available()) else '不可用')
print('显卡数量:',torch.cuda.device_count())
# print('是否支持BF16数字格式:','支持' if (torch.cuda.is_bf16_supported()) else '不支持')
print('当前显卡型号:',torch.cuda.get_device_name())
print('当前显卡的CUDA算力:',torch.cuda.get_device_capability())
print('当前显卡的总显存:',torch.cuda.get_device_properties(0).total_memory/1024/1024/1024,'GB')
print('是否支持TensorCore:','支持' if (torch.cuda.get_device_properties(0).major >= 7) else '不支持')
print('当前显卡的显存使用率:',torch.cuda.memory_allocated(0)/torch.cuda.get_device_properties(0).total_memory*100,'%')

输出示例

CUDA版本: 10.1
Pytorch版本: 1.7.1+cu101
显卡是否可用: 可用
显卡数量: 2
当前显卡型号: NVIDIA GeForce RTX 4080
当前显卡的CUDA算力: (8, 9)
当前显卡的总显存: 15.9915771484375 GB
是否支持TensorCore: 支持
当前显卡的显存使用率: 0.0 %