pynvml运行错误

环境

1. Windows 10（1903）
2. NVIDIA studio 456.38 驱动

问题

安装完驱动后发现 C:\Program Files\NVIDIA Corporation\NVSMI 里面是空的，所以会出现以下错误

Traceback (most recent call last):
  File "D:\Python\Python36\lib\site-packages\pynvml.py", line 641, in _LoadNvmlLibrary
    nvmlLib = CDLL(os.path.join(os.getenv("ProgramFiles", "C:/Program Files"), "NVIDIA Corporation/NVSMI/nvml.dll"))
  File "D:\Python\Python36\lib\ctypes\__init__.py", line 348, in __init__
    self._handle = _dlopen(self._name, mode)
OSError: [WinError 126] 找不到指定的模块。

During handling of the above exception, another exception occurred:

Traceback (most recent call last):
  File "H:\Code\PythonCode\PythonCode\gpu_info.py", line 2, in <module>
    pynvml.nvmlInit()
  File "D:\Python\Python36\lib\site-packages\pynvml.py", line 608, in nvmlInit
    _LoadNvmlLibrary()
  File "D:\Python\Python36\lib\site-packages\pynvml.py", line 646, in _LoadNvmlLibrary
    _nvmlCheckReturn(NVML_ERROR_LIBRARY_NOT_FOUND)
  File "D:\Python\Python36\lib\site-packages\pynvml.py", line 310, in _nvmlCheckReturn
    raise NVMLError(ret)
pynvml.NVMLError_LibraryNotFound: NVML Shared Library Not Found

解决方案

我们看到第一个错误是找不到nvml.dll，经过搜索之后发现在C:\Windows\System32\里有nvml.dll
将他复制到C:\Program Files\NVIDIA Corporation\NVSMI里再次执行程序
即可

运行的python程序

import pynvml
pynvml.nvmlInit()
# 这里的0是GPU id
handle = pynvml.nvmlDeviceGetHandleByIndex(0)
meminfo = pynvml.nvmlDeviceGetMemoryInfo(handle)
print(meminfo.used)
print(meminfo.total)