Hansimov's Blog

Appearance

Windows 安装支持 GPU 的 opencv-python ​

确保 Python 是 64 位 ​

sh
python -c "import sys,platform; print(sys.version); print(platform.architecture())"

安装 CUDA Toolkit + cuDNN ​

cuDNN 可能需要登录账号+填写个人信息才能下载。

CUDA 12.9 + cuDNN 9.10.2 ​

CUDA Toolkit 12.9 Downloads：

cuDNN 9.10.2：

CUDA 12.2 + cuDNN 8.9.3 ​

CUDA Toolkit 12.2 Downloads | NVIDIA Developer：

cuDNN v8.9.3 (July 11th, 2023), for CUDA 12.x：

点击安装 ​

CUDA 默认安装路径：

  • C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v12.x
  • 一开始会提示你解压位置，这个不是最终安装路径，可以选为 D:\CUDA12X，安装好后这个文件夹会自动删除
  • 为了不占用太多C盘空间，可以将安装路径设为：D:\CUDA-12.x
  • 安装好后，可以在系统环境变量的 PATH 中看到新增了 D:\CUDA-12.x\bin 和 D:\CUDA-12.x\libnvvp

cuDNN 默认安装路径：

  • C:\Program Files\NVIDIA\CUDNN\v9.x
  • 为了不占用太多C盘空间，可以将安装路径设为：D:\CUDNN-9.x
  • 需要把 Development, Runtime, Samples 都设为同一个路径
  • 解压后需要手动将文件复制到 CUDA 安装目录下对应的文件夹中

复制 cuDNN 文件到 CUDA 路径 ​

将下载好的 cuDNN 文件放到 CUDA 安装目录的对应路径：

  • CUDNN-9.10\bin\12.9\*.dll -> CUDA-12.9\bin\*.dll
  • CUDNN-9.10\include\12.9\*.h -> CUDA-12.9\include\*.h
  • CUDNN-9.10\lib\12.9\x64\*.lib -> CUDA-12.9\lib\x64\*.lib

卸载 CPU build 版本的 opencv-python ​

sh
pip uninstall -y opencv-python opencv-contrib-python
# pip uninstall -y opencv-python-headless opencv-contrib-python-headless
# pip uninstall -y opencv-python-rolling opencv_contrib_python_rolling

安装 CUDA 版本的 opencv-python ​

访问 cudawarped/opencv-python-cuda-wheels：

cuda 12.9 + cudnn 9.10.2：

cuda 12.2 + cudnn 8.9.3：

下载并安装：

sh
# cd <下载目录>
pip install opencv_contrib_python-4.12.0.88-cp37-abi3-win_amd64.whl

检查是否安装成功 ​

检查 OpenCV 版本信息：

sh
python -c "import cv2; print(cv2.getBuildInformation())"

检查是否支持 CUDA：

sh
python -c "import cv2; print('CUDA devices:', cv2.cuda.getCudaEnabledDeviceCount())"

输出形如：

sh
CUDA devices: 1