From 124eefecfe926fb37da06d007731012069190c4b Mon Sep 17 00:00:00 2001 From: jakirkham Date: Wed, 15 Jan 2025 09:29:37 -0800 Subject: [PATCH] Remove unneeded `smi` code --- .../cugraph/standalone/bulk_sampling/bench_cugraph_training.py | 2 -- 1 file changed, 2 deletions(-) diff --git a/benchmarks/cugraph/standalone/bulk_sampling/bench_cugraph_training.py b/benchmarks/cugraph/standalone/bulk_sampling/bench_cugraph_training.py index 60794654e4..66c34cc127 100644 --- a/benchmarks/cugraph/standalone/bulk_sampling/bench_cugraph_training.py +++ b/benchmarks/cugraph/standalone/bulk_sampling/bench_cugraph_training.py @@ -36,9 +36,7 @@ def init_pytorch_worker(rank: int, use_rmm_torch_allocator: bool = False) -> None: import cupy import rmm - from pynvml_utils.smi import nvidia_smi - smi = nvidia_smi.getInstance() pool_size = 16e9 # FIXME calculate this rmm.reinitialize(