Q

我可以在OpenCL中使用cuDNN嗎

2017-05-24 156 views 1 likes

1

我想安裝caffe，我想知道是否可以在AMD/OpenCL上使用cuDNN。因爲我的顯卡是AMD我可以在OpenCL中使用cuDNN嗎

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

2017-05-24 Snarcraft

A

回答

2

這恐怕是行不通的：cudnn是cuda的延伸，這是一個NVIDIA禮。因此，非NVIDIA GPU不支持CUDA，因此不支持cuDNN。
使用非NVIDIA顯卡時，您無法運行CUDA代碼（主要caffe分支），但您應該可以享受opencl GPU代碼。你應該給opencl分支一個機會。

2017-05-24 13:33:20 Shai

+0

好）謝謝！！我要去嘗試 – Snarcraft

1

簡短的回答是，如果你的顯卡是AMD，那麼你將不得不使用OpenCL，而不是cuDNN。你不能讓他們一起工作。

cuDNN和OpenCL都是competing libraries，所以嘗試並用它們甚至沒有意義。

如果您問是否可以在AMD硬件上使用NVIDIA的cuDNN庫，答案是否定的。它只是不兼容。 cuDNN專門用於NVIDIA硬件，並利用該芯片組的獨特性能。

2017-05-24 13:38:05 Wontonimo

+0

好吧）我明白了。謝謝 – Snarcraft

1

有cuDNN從英特爾的OpenCL的變種：

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

由於它是基於OpenCL的，也應該努力在AMD的GPU（雖然我沒有測試它自己）

2017-05-30 08:08:55 Ruyk

+0

好的）我會檢查，謝謝很多隊友:) – Snarcraft

+1

我認爲clDNN依賴於在非英特爾圖形中找不到的擴展。我看到內核正在使用子組（例如'intel_sub_group_block_read2'）。 – Tim

相關問題