search

plasma-umass / scalene

Scalene: a high-performance, high-precision CPU, GPU, and memory profiler for Python with AI-powered optimization proposals
Apache License 2.0
12.23k stars 399 forks source link

Fix CPU-only profiling failing with "AttributeError: module 'pynvml' … #838

Closed vavanade closed 4 months ago

vavanade commented 4 months ago

…has no attribute 'nvml'"

The error suppression does not work for an environment with nvidia-ml-py==12.555.43 and nvidia-ml-py3==7.352.0

emeryberger commented 4 months ago

Thanks for the PR! It was slightly incorrect but I've merged a corrected version.