python写个检测本机GPU的程序

要使用 Python 编写检测本机 GPU 的程序，可根据不同的 GPU 厂商和框架来实现。以下分别给出使用 NVIDIA GPU 的 CUDA 以及通用的 PyTorch 和 TensorFlow 框架的示例代码。 ### 使用 NVIDIA CUDA 若使用 NVIDIA GPU，可借助 `pynvml` 库来检测 GPU 信息。需先安装该库： ```bash pip install pynvml ``` 示例代码如下： ```python import pynvml # 初始化 NVML pynvml.nvmlInit() # 获取 GPU 设备数量 device_count = pynvml.nvmlDeviceGetCount() print(f"本机检测到的 GPU 数量: {device_count}") # 遍历每个 GPU 设备并打印信息 for i in range(device_count): handle = pynvml.nvmlDeviceGetHandleByIndex(i) info = pynvml.nvmlDeviceGetMemoryInfo(handle) name = pynvml.nvmlDeviceGetName(handle) print(f"GPU {i} 信息:") print(f" 名称: {name.decode('utf-8')}") print(f" 总显存: {info.total / 1024 ** 3:.2f} GB") print(f" 已使用显存: {info.used / 1024 ** 3:.2f} GB") print(f" 空闲显存: {info.free / 1024 ** 3:.2f} GB") # 关闭 NVML pynvml.nvmlShutdown() ``` ### 使用 PyTorch 若使用 PyTorch 框架，可使用以下代码检测 GPU 是否可用以及获取 GPU 信息： ```python import torch # 检查 GPU 是否可用 if torch.cuda.is_available(): print("GPU 可用") # 获取 GPU 设备数量 device_count = torch.cuda.device_count() print(f"本机检测到的 GPU 数量: {device_count}") # 打印每个 GPU 的名称 for i in range(device_count): print(f"GPU {i} 名称: {torch.cuda.get_device_name(i)}") else: print("GPU 不可用") ``` ### 使用 TensorFlow 若使用 TensorFlow 框架，可使用以下代码检测 GPU 是否可用： ```python import tensorflow as tf # 检查 GPU 是否可用 gpus = tf.config.list_physical_devices('GPU') if gpus: print("GPU 可用") print(f"本机检测到的 GPU 数量: {len(gpus)}") for gpu in gpus: print(f"GPU 信息: {gpu}") else: print("GPU 不可用") ```

创作声明：本文部分内容由AI辅助生成（AIGC），仅供参考

下一篇 python pandas删除df.column行后重置序列

目录

python写个检测本机GPU的程序

Python内容推荐

Python-监视GPU访问的Python脚本

Python摄像头实时点头检测 基于人脸关键点检测 使用pytorch实现

GPU编程实战-基于Python和CUDA.pdf

Python实时摄像头摇头检测代码 使用pytorch+opencv实现 基于人脸检测关键点

opencv4.2 python gpu编译版

单目摄像头车辆测距源码解析：Python转C语言实现，辅助驾驶中的深度学习技术测距方案参考陈光论文,基于单目摄像头的车辆测距源码（Python转C语言实现，支持GPU与CPU版本）,单目车辆测距源码

python 人脸检测demo

基于Python的车辆检测算法的实现.pdf

CnSTD-Python资源

opencv4.5.1-cuda10.1-python3.7.rar

基于python的带有关键点检测的超轻量级人脸检测器项目源码

Python-小型Python库自动将CUDAVISIBLEDEVICES设置为多GPU系统上最小负载的设备

GPU Computing in Python

Python实战应用代码-python_opencv实时视频目标检测

分别使用OpenCV、ONNXRuntime部署多任务的yolov5目标检测+语义分割，包含C++和Python两个版本的程序

Python-目标检测（yolov5-7.0安装及训练细胞）

使用 ONNXRuntime 部署 yolov5-lite 目标检测的 C++ 与 Python 双版本程序

python opencv车辆测速视频车速检测

基于Python搭建深度神经网络，实现花卉图像识别

Python基于pyCUDA实现GPU加速并行计算功能入门教程

详解python中GPU版本的opencv常用方法介绍

检测tensorflow是否使用gpu进行计算的方式

2025年扫路车行业大数据分析及市场预测

从MySQL迁移到Opengauss：SQL语法差异与兼容性实践手册

Java打包时提示‘无法访问xxx.class’，这通常是由哪些配置或结构问题导致的？

深度学习在生命科学中的革命性应用

告别MySQL依赖！手把手教你将Nacos 2.5.0的数据源切换到PostgreSQL（附完整建表SQL）

Java怎么用现有Word和Excel模板动态填充数据并导出？

微信PHP SDK资源包：开发者的必备工具

Linux命令行玩转多屏：dbus-send控制屏幕亮度+任务栏显示隐藏（附完整参数表）

Python摄像头实时点头检测基于人脸关键点检测使用pytorch实现

Python实时摄像头摇头检测代码使用pytorch+opencv实现基于人脸检测关键点