我是初学者,在pycharm 下,pytorch可以使用CUDA,能找到GPU设备,说明我NVIDIA 驱动没有问题。pip opencv-python 和pip opencv-contrib-python , 就找不到GPU的设备。会是哪里出问题了?
CUDA 版本是最新的12.3版本。 pip opencv-python 和pip opencv-contrib-python 也是最新的4.9080。
pychon 12版本,pychon7版本和pychon6版本都切换过了。都不行。
我使用的是windows系统。
pycharm 在opencv 上使用不了CUDA的问题。
- 写回答
- 好问题 0 提建议
- 追加酬金
- 关注问题
- 邀请回答
-
23条回答 默认 最新
关注 获得0.75元问题酬金 引自免费微信小程序:皆我百晓生
在PyCharm中配置OpenCV使用CUDA可能涉及到多个步骤,尤其是在确保OpenCV已经正确编译并支持CUDA的情况下。目前的OpenCV库(通过
pip install opencv-python
和pip install opencv-contrib-python
安装)默认情况下可能并未包含CUDA支持。要解决这个问题,请按照以下步骤排查:
-
确认OpenCV版本与CUDA兼容性: 首先需要确认你所安装的OpenCV版本是否真正支持CUDA 12.3。通常,你需要手动编译带有CUDA支持的OpenCV,因为pip提供的预编译包可能不包含CUDA支持。
-
手动编译OpenCV with CUDA支持: 从源代码编译OpenCV,并启用CUDA支持。以下是简化版的编译命令示例:
git clone https://github.com/opencv/opencv.git git clone https://github.com/opencv/opencv_contrib.git cd opencv mkdir build && cd build cmake -D CMAKE_BUILD_TYPE=RELEASE \ -D CMAKE_INSTALL_PREFIX=/usr/local \ -D OPENCV_ENABLE_NONFREE=ON \ -D INSTALL_PYTHON_EXAMPLES=ON \ -D INSTALL_C_EXAMPLES=OFF \ -D OPENCV_EXTRA_MODULES_PATH=../opencv_contrib/modules \ -D WITH_CUDA=ON \ -D CUDA_ARCH_BIN="7.5" \ # 根据你的GPU架构调整 -D WITH_CUDNN=ON \ -D ENABLE_FAST_MATH=1 \ -D CUDA_FAST_MATH=1 \ -D WITH_TBB=ON \ -D BUILD_opencv_python3=ON \ -D PYTHON_EXECUTABLE=$(which python) \ -D BUILD_EXAMPLES=ON .. make -j$(nproc) sudo make install
-
验证安装: 编译完成后,重新导入OpenCV模块并检查CUDA设备是否可被识别。例如:
import cv2 print(cv2.cuda.getCudaEnabledDeviceCount())
如果返回大于0的数字,则说明已成功连接到至少一个CUDA设备。
-
环境变量问题: 确保在运行PyCharm时,它使用的Python解释器能够访问到刚刚安装的CUDA支持的OpenCV库。
请注意,上述步骤可能需要根据你的具体环境进行调整,包括但不限于CUDA、CUDNN、Python版本以及系统环境等。如果你对上述步骤不熟悉,建议参考官方文档或相关教程进行操作。另外,务必确保系统的NVIDIA驱动程序、CUDA Toolkit和CUDNN都与你正在编译的OpenCV版本兼容。对于详细的编译指南,可以参考OpenCV官方文档。
解决 无用评论 打赏 举报-
悬赏问题
- ¥15 防火墙的混合模式配置
- ¥15 Ubuntu不小心注销了要怎么恢复啊
- ¥15 安装powerbuilder10卡在安装程序正在运行这个页面 没有下一步任何指令
- ¥15 关于mpi的问题:请问遇到这种情况需要怎么解决,出现这个问题后电脑不能进行mpi多核运行只能进行单核运行
- ¥50 微信聊天记录备份到电脑提示成功了,但还是没同步到电脑微信
- ¥15 python怎么在已有视频文件后添加新帧
- ¥20 虚幻UE引擎如何让多个同一个蓝图的NPC执行一样的动画,
- ¥15 fluent里模拟降膜反应的UDF编写
- ¥15 MYSQL 多表拼接link
- ¥15 关于某款2.13寸墨水屏的问题