The following transformer sample models are included in this repo to help you get started. These samples include both inference and training scripts, and you can either train the models from scratch or use the supplied pre-trained weights.
PyTorch with DirectML is supported on both the latest versions of Windows and the Windows Subsystem for Linux, and is available for download as a PyPI package. For more information about getting started with torch-directml, see our Windows or WSL 2 guidance on Microsoft Learn.
Once a Python (3.8 to 3.12) environment is setup, install the latest release of torch-directml by running the following command:
pip install torch-directml