Use DirectML for microsoft/phi-1_5
#24
by
shiqi1031
- opened
Not sure if same question has been asked, if it is possible to use DirectML (for AMD GPUs) on Windows to use phi-1.5?
Hello @shiqi1031 !
As far as I know, DirectML works with PyTorch 1.13 and 1.8: https://github.com/microsoft/DirectML/tree/master/PyTorch. However, you will need to double check if transformers works with these versions.
Regards,
Gustavo.
gugarosa
changed discussion status to
closed