我可以在OpenCL中使用cuDNN吗?

时间:2017-05-24 13:28:43

标签: opencl deep-learning caffe conv-neural-network cudnn

我正在尝试安装caffe,我想知道我是否可以在AMD / OpenCL中使用cuDNN。 因为我的显卡是AMD

https://github.com/BVLC/caffe/tree/opencl

4 个答案:

答案 0 :(得分:2)

我担心这不起作用:的延伸,这是NVIDIA的正当性。因此,非NVIDIA GPU不支持CUDA,因此不支持cuDNN 使用非NVIDIA卡,您无法运行CUDA代码(主要caffe分支),但您应该能够享受opencl GPU代码。你应该给opencl branch一个机会。

答案 1 :(得分:1)

简短的回答是,如果你的显卡是AMD,那么你必须使用OpenCL,而不是cuDNN。你不能让它们一起工作。

cuDNN和OpenCL是competition,因此尝试一起使用它们甚至没有意义。

如果您要问是否可以在AMD硬件上使用NVIDIA的cuDNN库,答案是否定的。它只是不兼容。 cuDNN专门用于NVIDIA硬件,并充分利用该芯片组的独特属性。

答案 2 :(得分:1)

英特尔的cuDNN有一个OpenCL变体:

https://github.com/01org/clDNN/

由于它是基于OpenCL的,所以也应该在AMD GPU上工作(虽然我自己没有测试过)

答案 3 :(得分:0)

恐怕您真的可以将其用于AMD显卡,因为clDNN是为DL推理而构建的,尤其是针对Intel显卡(HD和Iris)的。例如,检查OpenVINO toolkit(由Intel设计),它在GPU plugins的引擎盖下使用clDNN来加速对Intel GPU的推断。

在GPU插件页面上,它显示:

clDNN是用于深度学习(DL)应用程序的开源性能库,旨在加速基于英特尔®处理器图形(包括英特尔®HD图形和英特尔®Iris®图形)的深度学习推理。