pytorch-directml 1.8.0a0.dev220506

Tensors and Dynamic neural networks in Python with strong GPU acceleration

Stars: 1919, Watchers: 1919, Forks: 269, Open Issues: 186

The microsoft/DirectML repo was created 5 years ago and the last code push was 33 minutes ago.
The project is very popular with an impressive 1919 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 186 open issues on GitHub

• RuntimeError: don't know how to restore data location of torch.FloatStorage
• receiving this error
• Bump pillow from 7.0.0 to 9.0.0 in /TensorFlow/squeezenet/src
• Could not run 'aten::_embedding_bag' on RX 6900 XT
• Will Pytorch-DirectML support automatic mixed precision in the future?
• How to use pytorch-directml in torch.nn?
• importlib_metadata.version("torch") does not work
• resnet50 tutorial failed for me
• argmax function cause error (pytorch-directml)
• .to('dml') not working
• Could not run 'aten::index_select' on RX 6700
• onnxruntime-directml slower than onnxruntime cpu
• [pytorch-dml] Torch Transfomer
• Why does directml not focus on onnxruntime but tensorflow?
• Bump opencv-python from 4.1.1.26 to 4.2.0.32 in /TensorFlow/yolov3

See more issues on GitHub