CTranslate2 DirectML支持

7y4bm7vi  于 2个月前  发布在  其他
关注(0)|答案(2)|浏览(45)

Assume that I already followed Microsoft's instructions to Enable PyTorch with DirectML on Windows and the DirectML library loads correctly according to MS's example code.
If I wanted to use PyTorch + DirectML + Ctranslate2 + an AMD GPU, how would I do it? Does Ctranslate2 need any major updates? Shouldn't I just need to pass directml instead of cuda to device= as found in the quickstart ?
What is the bottleneck to this not working right now?
Aside: MS's instructions for DirectML's Python library install pytorch==2.0 as of early 2024. MS's documentation suggests it only works with PyTorch 1.3, which seems to be incorrect? So, it should not be a matter of supporting PyTorch 1.x since DirectML runs on 2.x in practice.

kmpatx3s

kmpatx3s1#

我也需要DirectML。

agyaoht7

agyaoht72#

目前,Ctranslate2不支持DirectML。要支持这个,需要为这个后端提供新的实现。

相关问题