pynvml模块使用方法

最近有需要使用nvml调用底层显卡信息的需求,所以查了下python有相关的模块pynvml可以使用,下面就直接看看一些基本使用的方法首先我们安装这个模块pip install nvidia-ml-py3然后使用时我们需要初始化该方法import pynvmlpynvml.nvmlInit()使用外以后也需要进行关闭,类似于析构方法pynvml.nvmlShutdown()然后下面是收集的一些常用方法模块#!/usr/bin/python3#pip install nvidia-ml-py3i...

阅读全文