pytorch-directml 1.8.0a0.dev220506

Tensors and Dynamic neural networks in Python with strong GPU acceleration

Stars: 2546, Watchers: 2546, Forks: 333, Open Issues: 12

The microsoft/DirectML repo was created 6 years ago and the last code push was 6 days ago.
The project is very popular with an impressive 2546 github stars!

How to Install pytorch-directml

You can install pytorch-directml using pip

pip install pytorch-directml

or add it to a project with poetry

poetry add pytorch-directml

Package Details

Author

Microsoft Corporation

License

BSD-3

Homepage

https://github.com/microsoft/directml

PyPi:

https://pypi.org/project/pytorch-directml/

GitHub Repo:

https://github.com/microsoft/directml

Classifiers

• Scientific/Engineering
• Scientific/Engineering/Artificial Intelligence
• Scientific/Engineering/Mathematics
• Software Development
• Software Development/Libraries
• Software Development/Libraries/Python Modules

Related Packages

Errors

A list of common pytorch-directml errors.

Code Examples

Here are some pytorch-directml code examples and snippets.

GitHub Issues

The pytorch-directml package has 12 open issues on GitHub

• Bump protobuf from 3.11.3 to 7.34.0rc1 in /TensorFlow/TF1/squeezenet/src
• "Invalid Argument (80070057)" crash with Transformer-based models (D-FINE/RT-DETR) due to Int64 indices in Gather/Scatter ops
• Bump werkzeug from 1.0.0 to 3.1.5 in /TensorFlow/TF1/squeezenet/src
• Bump urllib3 from 1.25.8 to 2.6.3 in /TensorFlow/TF1/squeezenet/src
• "Invalid device_id argument supplied" -> typo
• Access to latest defines

See more issues on GitHub