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)
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值