Halcon深度学习用不了GPU训练和推理

Gary88 · 发表于 2024-2-20 12:21:34

本帖最后由 Gary88 于 2024-2-20 12:22 编辑

硬件3080显卡，软件Halcon23.11 + CUDA11.5+Cudnn8.9 运行深度学习例程总是只用了CPU，没有用GPU，请问大神是什么原因，谢谢

绝地武士 · 发表于 2024-2-20 18:12:27

能读取到GPU参数么？如果不能那就是显卡驱动的问题，尝试更新看看

ybvip · 发表于 2024-2-21 10:58:44

未正确安装CUDA和cuDNN：确保已正确安装了CUDA和cuDNN，并且安装的版本与你的Halcon23.11兼容。如果版本不匹配，可能会导致GPU无法被正确识别或使用。

未正确配置环境变量：在安装CUDA和cuDNN后，需要设置相应的环境变量，以便系统可以找到相应的库和头文件。确保你已正确设置了CUDA_HOME和PATH等环境变量。

代码中未明确指定GPU设备：有些深度学习框架需要在代码中明确指定使用GPU，否则默认使用CPU。检查你的代码，确保已将模型和计算移动到GPU设备上。

驱动问题：确保你的显卡驱动已正确安装，并且与CUDA和cuDNN兼容。

figlio · 发表于 2024-3-30 16:51:22

23.11好像确实有点问题，无法识别gpu