环境
- Windows 11 23H2
- conda 23.11.0
- PyTorch 2.1.2 py3.11_cuda12.1_cudnn8_0
- OpenCV-contrib With-Cuda 4.9.0.80
问题情形
PyTorch由conda安装,随包附带一套Cuda环境。
后续需要在这个环境中安装OpenCV的Cuda加速版本,选择从opencv-python-cuda-wheels-4.9.0.80处下载编译好的whl文件,并直接使用pip install
指令安装。(环境中原本没有标准OpenCV发行包)
在python中执行import cv2
时,报错:
Traceback (most recent call last):
File “”, line 1, in
File “C:\ProgramData\miniconda3\envs\mm\Lib\site-packages\cv2_init_.py”, line 181, in
bootstrap()
File “C:\ProgramData\miniconda3\envs\mm\Lib\site-packages\cv2_init_.py”, line 153, in bootstrap
native_module = importlib.import_module(“cv2”)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File “C:\ProgramData\miniconda3\envs\mm\Lib\importlib_init_.py”, line 126, in import_module
return _bootstrap._gcd_import(name[level:], package, level)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
ImportError: DLL load failed while importing cv2: 找不到指定的模块。
问题分析
OpenCV的CUDA加速版本需要CUDA Toolkit、CUDA Runtime、CuDNN等组件以正常运行。由于我的CUDA环境是跟随pytorch包安装时自动绑定在一起install的,也就是pytorch官方帮忙做了一个pytorch+cuda的包,以便于install。但是这样安装的cuda环境并不属于conda的整个虚拟环境,而是只属于PyTorch自身。因此在PyTorch外部尝试调用需要CUDA环境的包时,会找不到相应的支持DLL
临时解决办法
手动为python添加DLL目录,将PyTorch自带的CUDA环境的DLL也添加到python的运行环境中去。
import os
os.add_dll_directory(r"......\envs\YOURENV\Lib\site-packages\torch\lib")
长期解决办法
思路是将所需要的DLL文件直接复制到Conda环境中专门存储所有DLL文件的目录下就可以了
Source:miniconda3\envs\YOURENV\Lib\site-packages\torch\lib\cudnn64_8.dll
Destination:miniconda3\envs\YOURENV\bin\cudnn64_8.dll
验证
import cv2
cv2.cuda.getCudaEnabledDeviceCount() # 1