解决 pynvml 报错 NVML Shared Library Not Found / WinError 126

爱被打了一巴掌 2022-12-19 11:38 288阅读 0赞

> pynvml包直接在Windows系统上安装可能会报出模块找不到的错误，本文记录解决方法。

### 问题复现 ###

> 在Windows中调用 `pynvml.nvmlInit()`时出错：

![418dd5bd07dec6f4cbd1ad61160d5ef4.png][]

> 错误信息：

File "E:\Program_Files\Annoconda\lib\site-packages\pynvml\nvml.py", line 731, in _load_nvml_library
        nvml_lib = CDLL(os.path.join(os.getenv("ProgramFiles", "C:/Program Files"), "NVIDIA Corporation/NVSMI/nvml.dll"))
      File "E:\Program_Files\Annoconda\lib\ctypes\__init__.py", line 356, in __init__
        self._handle = _dlopen(self._name, mode)
    OSError: [WinError 126] 找不到指定的模块。

### 问题分析 ###

*  在源代码文件 numl.py 中 731 行附近：

if (sys.platform[:3] == "win"):
        # cdecl calling convention
        # load nvml.dll from %ProgramFiles%/NVIDIA Corporation/NVSMI/nvml.dll
        nvml_lib = CDLL(os.path.join(os.getenv("ProgramFiles", "C:/Program Files"), "NVIDIA Corporation/NVSMI/nvml.dll"))
    else:
        # assume linux
        nvml_lib = CDLL("libnvidia-ml.so.1")

> 可以看到模块在尝试寻找 `C:/Program Files/NVIDIA Corporation/NVSMI/nvml.dll` 文件，如果你的`nvml.dll`文件不在这就会报错。

### 解决方案 ###

> 知道问题的原因，排除故障有两种方案，但都需要知道当前的`vnml.dll`文件在哪。

##### 确定 nvidia-smi.exe 位置 #####

*  打开cmd，输入命令：

where nvidia-smi

> 会返回当前系统的 nvidia-smi.exe的位置 （前提是当前cuda安装正常，可以使用nvidia-smi）

![e85a782461113f27d74af5197424a033.png][]

*  去相应位置查看，发现`vnml.dll`文件和他在一起，所以只要将代码中路径指向它就行了。

#### 方案1： 修改源码： ####

*  修改`numl.py` 中第 731 行代码：

nvml_lib = CDLL(os.path.join(os.getenv("ProgramFiles", "C:/Program Files"), "NVIDIA Corporation/NVSMI/nvml.dll"))

改为你自己的路径：

nvml_lib = CDLL(r"C:\Windows\System32\nvml.dll")

> 注意反斜杠路径需要在前面加 `r`

#### 方案2： 为源码创建运行环境 ####

*  源码不会出错，错的是我们自己的环境，改过来就是了。
 *  创建文件夹`C:\Program Files\NVIDIA Corporation\NVSMI`
 *  将`nvml.dll`文件放在文件夹里

#### 运行示例 ####

![c65c8b38afa812b278b28a45de6cecaf.png][]

### 参考资料： ###

*  https://blog.csdn.net/com\_fang\_bean/article/details/107988018

[418dd5bd07dec6f4cbd1ad61160d5ef4.png]: /images/20221120/6889fdfdff2043ce91aedfcec5d8668e.png
[e85a782461113f27d74af5197424a033.png]: /images/20221120/34d109588ede4e0a89fd523aff5691c3.png
[c65c8b38afa812b278b28a45de6cecaf.png]: /images/20221120/04c5822db007449595526bc72b04cea6.png