The following computer vision sample models are included in this repo to help you get started. The sample 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