# PyTorch with DirectML
DirectML acceleration for PyTorch is currently available for Public Preview. PyTorch with DirectML enables training and inference of complex machine learning models on a wide range of DirectX 12-compatible hardware.
DirectML is a high-performance, hardware-accelerated DirectX 12 library for machine learning. DirectML provides GPU acceleration for common machine learning tasks across a broad range of supported hardware and drivers, including all DirectX 12-capable GPUs from vendors such as AMD, Intel, NVIDIA, and Qualcomm.
More information about DirectML can be found in [Introduction to DirectML](https://docs.microsoft.com/windows/win32/direct3d12/dml-intro).
PyTorch on DirectML is supported on both the latest versions of Windows 10 and the [Windows Subsystem for Linux](https://docs.microsoft.com/windows/wsl/about), and is available for download as a PyPI package. For more information about getting started, see [GPU accelerated ML training (docs.microsoft.com)](http://aka.ms/gpuinwsldocs)
## Samples
Refer to the [Pytorch with DirectML Samples Repo](https://github.com/microsoft/DirectML/tree/master/PyTorch) for samples.
## Roadmap
`torch-directml` is actively under development and we're always adding more operators. For a list of all the operators we support and their data type coverage, refer to the [PyTorch DirectML Operator Roadmap](https://github.com/microsoft/DirectML/wiki/PyTorch-DirectML-Operator-Roadmap) in the DirectML repository wiki. If you require support for an operator that isn't in this list, see the [Feedback](#feedback) section below on how to file an issue.
## Feedback
We look forward to hearing from you!
* For TensorFlow with DirectML issues, bugs, and feedback; or for general DirectML issues and feedback, please [file an issue](https://github.com/microsoft/DirectML-Samples/issues) or contact us directly at askdirectml@microsoft.com.
* For PyTorch with DirectML issues, bugs, and feedback; or for general DirectML issues and feedback, please [file an issue](https://github.com/microsoft/DirectML-Samples/issues) or contact us directly at askdirectml@microsoft.com.
* For Windows ML issues, please file a GitHub issue at [microsoft/Windows-Machine-Learning](https://github.com/Microsoft/Windows-Machine-Learning/issues) or contact us directly at askwindowsml@microsoft.com.
* For ONNX Runtime issues, please file an issue at [microsoft/onnxruntime](https://github.com/microsoft/onnxruntime/issues).
## Contributing
This project welcomes contributions and suggestions. Most contributions require you to agree to a
Contributor License Agreement (CLA) declaring that you have the right to, and actually do, grant us
the rights to use your contribution. For details, visit https://cla.microsoft.com.
When you submit a pull request, a CLA-bot will automatically determine whether you need to provide
a CLA and decorate the PR appropriately (e.g., label, comment). Simply follow the instructions
provided by the bot. You will only need to do this once across all repos using our CLA.
This project has adopted the [Microsoft Open Source Code of Conduct](https://opensource.microsoft.com/codeofconduct/).
For more information see the [Code of Conduct FAQ](https://opensource.microsoft.com/codeofconduct/faq/) or
contact [opencode@microsoft.com](mailto:opencode@microsoft.com) with any additional questions or comments.
## Data Collection Notice
The software may collect information about you and your use of the software and send it to Microsoft. Microsoft may use this information to provide services and improve our products and services. There are also some features in the software that may enable you and Microsoft to collect data from users of your applications. If you use these features, you must comply with applicable law, including providing appropriate notices to users of your applications together with a copy of Microsoft's privacy statement. Our privacy statement is located at https://go.microsoft.com/fwlink/?LinkID=824704. You can learn more about data collection and use in the help documentation and our privacy statement. Your use of the software operates as your consent to these practices.
Specifically, in torch-directml, we are collecting the GPU device info and operators that fall back to CPU for improving operator coverage.
###
## External Links
- [PyTorch homepage](https://pytorch.org/)
- [PyTorch DirectML Operator Roadmap](https://github.com/microsoft/DirectML/wiki/PyTorch-DirectML-Operator-Roadmap)
- [Pytorch with DirectML Samples Repo](https://github.com/microsoft/DirectML/tree/master/PyTorch)
Raw data
{
"_id": null,
"home_page": "https://github.com/microsoft/directml",
"name": "torch-directml",
"maintainer": null,
"docs_url": null,
"requires_python": ">=3.7",
"maintainer_email": null,
"keywords": "pytorch, plugin, machine learning, directml, directx, d3d12",
"author": "Microsoft Corporation",
"author_email": "askdirectml@microsoft.com",
"download_url": null,
"platform": null,
"description": "# PyTorch with DirectML\nDirectML acceleration for PyTorch is currently available for Public Preview. PyTorch with DirectML enables training and inference of complex machine learning models on a wide range of DirectX 12-compatible hardware.\n\nDirectML is a high-performance, hardware-accelerated DirectX 12 library for machine learning. DirectML provides GPU acceleration for common machine learning tasks across a broad range of supported hardware and drivers, including all DirectX 12-capable GPUs from vendors such as AMD, Intel, NVIDIA, and Qualcomm.\n\nMore information about DirectML can be found in [Introduction to DirectML](https://docs.microsoft.com/windows/win32/direct3d12/dml-intro).\n\nPyTorch on DirectML is supported on both the latest versions of Windows 10 and the [Windows Subsystem for Linux](https://docs.microsoft.com/windows/wsl/about), and is available for download as a PyPI package. For more information about getting started, see [GPU accelerated ML training (docs.microsoft.com)](http://aka.ms/gpuinwsldocs)\n\n\n## Samples\nRefer to the [Pytorch with DirectML Samples Repo](https://github.com/microsoft/DirectML/tree/master/PyTorch) for samples. \n\n## Roadmap\n`torch-directml` is actively under development and we're always adding more operators. For a list of all the operators we support and their data type coverage, refer to the [PyTorch DirectML Operator Roadmap](https://github.com/microsoft/DirectML/wiki/PyTorch-DirectML-Operator-Roadmap) in the DirectML repository wiki. If you require support for an operator that isn't in this list, see the [Feedback](#feedback) section below on how to file an issue. \n\n## Feedback\n\nWe look forward to hearing from you!\n\n* For TensorFlow with DirectML issues, bugs, and feedback; or for general DirectML issues and feedback, please [file an issue](https://github.com/microsoft/DirectML-Samples/issues) or contact us directly at askdirectml@microsoft.com.\n\n* For PyTorch with DirectML issues, bugs, and feedback; or for general DirectML issues and feedback, please [file an issue](https://github.com/microsoft/DirectML-Samples/issues) or contact us directly at askdirectml@microsoft.com.\n\n* For Windows ML issues, please file a GitHub issue at [microsoft/Windows-Machine-Learning](https://github.com/Microsoft/Windows-Machine-Learning/issues) or contact us directly at askwindowsml@microsoft.com.\n\n* For ONNX Runtime issues, please file an issue at [microsoft/onnxruntime](https://github.com/microsoft/onnxruntime/issues).\n\n## Contributing\n\nThis project welcomes contributions and suggestions. Most contributions require you to agree to a\nContributor License Agreement (CLA) declaring that you have the right to, and actually do, grant us\nthe rights to use your contribution. For details, visit https://cla.microsoft.com.\n\nWhen you submit a pull request, a CLA-bot will automatically determine whether you need to provide\na CLA and decorate the PR appropriately (e.g., label, comment). Simply follow the instructions\nprovided by the bot. You will only need to do this once across all repos using our CLA.\n\nThis project has adopted the [Microsoft Open Source Code of Conduct](https://opensource.microsoft.com/codeofconduct/).\nFor more information see the [Code of Conduct FAQ](https://opensource.microsoft.com/codeofconduct/faq/) or\ncontact [opencode@microsoft.com](mailto:opencode@microsoft.com) with any additional questions or comments.\n\n## Data Collection Notice\n\nThe software may collect information about you and your use of the software and send it to Microsoft. Microsoft may use this information to provide services and improve our products and services. There are also some features in the software that may enable you and Microsoft to collect data from users of your applications. If you use these features, you must comply with applicable law, including providing appropriate notices to users of your applications together with a copy of Microsoft's privacy statement. Our privacy statement is located at https://go.microsoft.com/fwlink/?LinkID=824704. You can learn more about data collection and use in the help documentation and our privacy statement. Your use of the software operates as your consent to these practices.\n\nSpecifically, in torch-directml, we are collecting the GPU device info and operators that fall back to CPU for improving operator coverage.\n### \n\n\n## External Links\n\n- [PyTorch homepage](https://pytorch.org/)\n- [PyTorch DirectML Operator Roadmap](https://github.com/microsoft/DirectML/wiki/PyTorch-DirectML-Operator-Roadmap)\n- [Pytorch with DirectML Samples Repo](https://github.com/microsoft/DirectML/tree/master/PyTorch)\n\n",
"bugtrack_url": null,
"license": "MIT",
"summary": "A DirectML backend for hardware acceleration in PyTorch.",
"version": "0.2.5.dev240914",
"project_urls": {
"Bug Tracker": "https://github.com/microsoft/directml/issues",
"Homepage": "https://github.com/microsoft/directml"
},
"split_keywords": [
"pytorch",
" plugin",
" machine learning",
" directml",
" directx",
" d3d12"
],
"urls": [
{
"comment_text": "",
"digests": {
"blake2b_256": "a0ad14e084f9f4882cdae86f2773fbf95807f56a282b5aea9df8048302491b62",
"md5": "b309f79c86b19f26da21ba1d857da0fa",
"sha256": "30e29872f4d6059dd784897fa2df0cddb80f3874ed4f0860758790286e95a823"
},
"downloads": -1,
"filename": "torch_directml-0.2.5.dev240914-cp310-cp310-manylinux2010_x86_64.whl",
"has_sig": false,
"md5_digest": "b309f79c86b19f26da21ba1d857da0fa",
"packagetype": "bdist_wheel",
"python_version": "cp310",
"requires_python": ">=3.7",
"size": 24949002,
"upload_time": "2024-09-15T13:20:54",
"upload_time_iso_8601": "2024-09-15T13:20:54.738215Z",
"url": "https://files.pythonhosted.org/packages/a0/ad/14e084f9f4882cdae86f2773fbf95807f56a282b5aea9df8048302491b62/torch_directml-0.2.5.dev240914-cp310-cp310-manylinux2010_x86_64.whl",
"yanked": false,
"yanked_reason": null
},
{
"comment_text": "",
"digests": {
"blake2b_256": "8ef9a188cc5e989881b98d3c3d8170ad95c783def75f2d544cebcb98507549fc",
"md5": "359f539c224cd35f62fbb83768d1759a",
"sha256": "9fe494ffd2c8ab9079f13404d052fc261cd8efb639a776c9075e58d9c64d6cb2"
},
"downloads": -1,
"filename": "torch_directml-0.2.5.dev240914-cp310-cp310-win_amd64.whl",
"has_sig": false,
"md5_digest": "359f539c224cd35f62fbb83768d1759a",
"packagetype": "bdist_wheel",
"python_version": "cp310",
"requires_python": ">=3.7",
"size": 8954792,
"upload_time": "2024-09-15T13:20:58",
"upload_time_iso_8601": "2024-09-15T13:20:58.099102Z",
"url": "https://files.pythonhosted.org/packages/8e/f9/a188cc5e989881b98d3c3d8170ad95c783def75f2d544cebcb98507549fc/torch_directml-0.2.5.dev240914-cp310-cp310-win_amd64.whl",
"yanked": false,
"yanked_reason": null
},
{
"comment_text": "",
"digests": {
"blake2b_256": "b582185f419c642751396b32195d352fd2410594c4401011d2b33c0399417f2c",
"md5": "dccb525d59d5715a5a6fbe97ca391c02",
"sha256": "6afb675585d30018c813e5ba203a3437073748919af8ab3e910092a0e0ec531f"
},
"downloads": -1,
"filename": "torch_directml-0.2.5.dev240914-cp311-cp311-manylinux2010_x86_64.whl",
"has_sig": false,
"md5_digest": "dccb525d59d5715a5a6fbe97ca391c02",
"packagetype": "bdist_wheel",
"python_version": "cp311",
"requires_python": ">=3.7",
"size": 24950103,
"upload_time": "2024-09-15T13:21:03",
"upload_time_iso_8601": "2024-09-15T13:21:03.029523Z",
"url": "https://files.pythonhosted.org/packages/b5/82/185f419c642751396b32195d352fd2410594c4401011d2b33c0399417f2c/torch_directml-0.2.5.dev240914-cp311-cp311-manylinux2010_x86_64.whl",
"yanked": false,
"yanked_reason": null
},
{
"comment_text": "",
"digests": {
"blake2b_256": "848b00528e6c75e030cc5f1fc1d08c58c46ecdbec9cd406b1dfd03023e3af4aa",
"md5": "d9edeef337e1163100f814848cad3f84",
"sha256": "3315b6c7e898685827607f1d8170dacc386ac248502aba9bd36cf82e78d930bb"
},
"downloads": -1,
"filename": "torch_directml-0.2.5.dev240914-cp311-cp311-win_amd64.whl",
"has_sig": false,
"md5_digest": "d9edeef337e1163100f814848cad3f84",
"packagetype": "bdist_wheel",
"python_version": "cp311",
"requires_python": ">=3.7",
"size": 8955958,
"upload_time": "2024-09-15T13:21:06",
"upload_time_iso_8601": "2024-09-15T13:21:06.539680Z",
"url": "https://files.pythonhosted.org/packages/84/8b/00528e6c75e030cc5f1fc1d08c58c46ecdbec9cd406b1dfd03023e3af4aa/torch_directml-0.2.5.dev240914-cp311-cp311-win_amd64.whl",
"yanked": false,
"yanked_reason": null
},
{
"comment_text": "",
"digests": {
"blake2b_256": "47e1636e6bc0fbf7bf13a2a206f7b47ecce381335c90c0a761bac243db6c2a37",
"md5": "6079c063c8b79fe9f4cb0e83cbc172c6",
"sha256": "26915aff5008a8567ea7641b74cf8cb53c1767d0c7163fc06e0a587e7c1c9dce"
},
"downloads": -1,
"filename": "torch_directml-0.2.5.dev240914-cp312-cp312-manylinux2010_x86_64.whl",
"has_sig": false,
"md5_digest": "6079c063c8b79fe9f4cb0e83cbc172c6",
"packagetype": "bdist_wheel",
"python_version": "cp312",
"requires_python": ">=3.7",
"size": 24949394,
"upload_time": "2024-09-15T13:21:10",
"upload_time_iso_8601": "2024-09-15T13:21:10.988986Z",
"url": "https://files.pythonhosted.org/packages/47/e1/636e6bc0fbf7bf13a2a206f7b47ecce381335c90c0a761bac243db6c2a37/torch_directml-0.2.5.dev240914-cp312-cp312-manylinux2010_x86_64.whl",
"yanked": false,
"yanked_reason": null
},
{
"comment_text": "",
"digests": {
"blake2b_256": "71f2eb7d3c601310c758f6aef891773e4a00f7388eeb613d536c0d2681fcb105",
"md5": "410852994b55131ba86f3fd4f301d86c",
"sha256": "ea19d11e33e9450b290311c06f7eb10924dd25c555e504d367b7b437d3eb24d0"
},
"downloads": -1,
"filename": "torch_directml-0.2.5.dev240914-cp312-cp312-win_amd64.whl",
"has_sig": false,
"md5_digest": "410852994b55131ba86f3fd4f301d86c",
"packagetype": "bdist_wheel",
"python_version": "cp312",
"requires_python": ">=3.7",
"size": 8956338,
"upload_time": "2024-09-15T13:21:13",
"upload_time_iso_8601": "2024-09-15T13:21:13.901587Z",
"url": "https://files.pythonhosted.org/packages/71/f2/eb7d3c601310c758f6aef891773e4a00f7388eeb613d536c0d2681fcb105/torch_directml-0.2.5.dev240914-cp312-cp312-win_amd64.whl",
"yanked": false,
"yanked_reason": null
},
{
"comment_text": "",
"digests": {
"blake2b_256": "f79978828ee9b10ae7548fe77b8b11820a17868cb1a41379443dcdbe534587b8",
"md5": "b15b9c6ab6560858113c3bd5a58b2a0f",
"sha256": "9b263c7d88ea225ce35b116441e1585fc753e9d69f64a8b6d83aabdc6c511517"
},
"downloads": -1,
"filename": "torch_directml-0.2.5.dev240914-cp38-cp38-manylinux2010_x86_64.whl",
"has_sig": false,
"md5_digest": "b15b9c6ab6560858113c3bd5a58b2a0f",
"packagetype": "bdist_wheel",
"python_version": "cp38",
"requires_python": ">=3.7",
"size": 66401920,
"upload_time": "2024-09-15T13:21:20",
"upload_time_iso_8601": "2024-09-15T13:21:20.583687Z",
"url": "https://files.pythonhosted.org/packages/f7/99/78828ee9b10ae7548fe77b8b11820a17868cb1a41379443dcdbe534587b8/torch_directml-0.2.5.dev240914-cp38-cp38-manylinux2010_x86_64.whl",
"yanked": false,
"yanked_reason": null
},
{
"comment_text": "",
"digests": {
"blake2b_256": "b4d133c35c3017bc1e2236c20a3183fb874b0fe77309c788f9d053555cb7c4e1",
"md5": "0b5d0c9df63a7cdf8c494cca306949ce",
"sha256": "6e18fb706d15cc6d0d3de49f46a7edc07ae669531c851d7d8f98855f4974f9e6"
},
"downloads": -1,
"filename": "torch_directml-0.2.5.dev240914-cp38-cp38-win_amd64.whl",
"has_sig": false,
"md5_digest": "0b5d0c9df63a7cdf8c494cca306949ce",
"packagetype": "bdist_wheel",
"python_version": "cp38",
"requires_python": ">=3.7",
"size": 8954967,
"upload_time": "2024-09-15T13:21:24",
"upload_time_iso_8601": "2024-09-15T13:21:24.415105Z",
"url": "https://files.pythonhosted.org/packages/b4/d1/33c35c3017bc1e2236c20a3183fb874b0fe77309c788f9d053555cb7c4e1/torch_directml-0.2.5.dev240914-cp38-cp38-win_amd64.whl",
"yanked": false,
"yanked_reason": null
},
{
"comment_text": "",
"digests": {
"blake2b_256": "2886746d81e55c8786ca8350224f2aacfb58c8e816d9cad73b90245859a48a2a",
"md5": "61fe25e5de85c296f0000e44ca51bf9d",
"sha256": "9c1988e95694520b31f1056bc038b76a62c9ca63f2d4e93f1abea1d293ec49ce"
},
"downloads": -1,
"filename": "torch_directml-0.2.5.dev240914-cp39-cp39-manylinux2010_x86_64.whl",
"has_sig": false,
"md5_digest": "61fe25e5de85c296f0000e44ca51bf9d",
"packagetype": "bdist_wheel",
"python_version": "cp39",
"requires_python": ">=3.7",
"size": 24949135,
"upload_time": "2024-09-15T13:21:28",
"upload_time_iso_8601": "2024-09-15T13:21:28.620554Z",
"url": "https://files.pythonhosted.org/packages/28/86/746d81e55c8786ca8350224f2aacfb58c8e816d9cad73b90245859a48a2a/torch_directml-0.2.5.dev240914-cp39-cp39-manylinux2010_x86_64.whl",
"yanked": false,
"yanked_reason": null
},
{
"comment_text": "",
"digests": {
"blake2b_256": "a3bb1e9b55d1f56d899777e157322a585114b20b18888c05ddf23f2d48e91ee7",
"md5": "75cd9ae4284bae8828db34aaff3a8fcf",
"sha256": "488509f0e8deb22f052b56f5cdad3a55878b65a7d99ee4b448fb4ab3cbb8d8ea"
},
"downloads": -1,
"filename": "torch_directml-0.2.5.dev240914-cp39-cp39-win_amd64.whl",
"has_sig": false,
"md5_digest": "75cd9ae4284bae8828db34aaff3a8fcf",
"packagetype": "bdist_wheel",
"python_version": "cp39",
"requires_python": ">=3.7",
"size": 8954122,
"upload_time": "2024-09-15T13:21:31",
"upload_time_iso_8601": "2024-09-15T13:21:31.948806Z",
"url": "https://files.pythonhosted.org/packages/a3/bb/1e9b55d1f56d899777e157322a585114b20b18888c05ddf23f2d48e91ee7/torch_directml-0.2.5.dev240914-cp39-cp39-win_amd64.whl",
"yanked": false,
"yanked_reason": null
}
],
"upload_time": "2024-09-15 13:20:54",
"github": true,
"gitlab": false,
"bitbucket": false,
"codeberg": false,
"github_user": "microsoft",
"github_project": "directml",
"travis_ci": false,
"coveralls": false,
"github_actions": false,
"lcname": "torch-directml"
}