pytorch-directml

Tensors and Dynamic neural networks in Python with strong GPU acceleration

Installation

In a virtualenv (see these instructions if you need to create one):

pip3 install pytorch-directml

Releases

Version Released Bullseye
Python 3.9
Bookworm
Python 3.11
Files
1.8.0a0.dev220506 pre-release yanked 2022-05-24
1.8.0a0.dev220224 pre-release yanked 2022-02-24
1.8.0a0.dev211021 pre-release yanked 2021-10-21

Issues with this package?

Page last updated 2025-06-27 20:02:07 UTC