Skip to content

Instantly share code, notes, and snippets.

#!/usr/bin/env python
from __future__ import print_function
from nvidia.pynvml import *
import socket
import time
HOSTNAME = socket.gethostname().split(".")[0]
try:
nvmlInit()