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.
2条答案
按热度按时间kmpatx3s1#
我也需要DirectML。
agyaoht72#
目前,Ctranslate2不支持
DirectML
。要支持这个,需要为这个后端提供新的实现。