Jing Zhang vinjn

AI Infra in NVIDIA

vinjn / nsight.sh

Created December 12, 2023 06:41 — forked from mcarilli/nsight.sh

Favorite nsight systems profiling commands for Pytorch scripts

	# This isn't supposed to run as a bash script, i named it with ".sh" for syntax highlighting.

	# https://developer.nvidia.com/nsight-systems
	# https://docs.nvidia.com/nsight-systems/profiling/index.html

	# My preferred nsys (command line executable used to create profiles) commands
	#
	# In your script, write
	# torch.cuda.nvtx.range_push("region name")
	# ...

vinjn / slurm-gpu-stats.py

Created December 18, 2024 03:30 — forked from schmmd/slurm-gpu-stats.py

Slurm GPU Stats

	#!/usr/bin/env python

	import os
	import re
	import subprocess
	import socket
	import sys

	def pids_of_jid(jid):
	result = subprocess.run(["sstat", "-p", "--format=PID", "-j", jid, "--noheader"], stdout=subprocess.PIPE)

vinjn / index.html

Created June 5, 2025 18:20 — forked from chromy/index.html

	<!doctype html>
	<html lang="en-us">
	<link rel="shortcut icon" href="data:image/x-icon;," type="image/x-icon">
	<body>

	<style>
	pre {
	border: 1px solid #eee;
	margin: 10px 0;
	font-family: monospace;