Conda+PyTorch+OpenCV-contrib-cuda环境下，import cv2 出现dll找不到的问题

环境

• 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