Installation

Pip

ml4gw is installable with pip:

$ pip install ml4gw

To build with a specific version of PyTorch/CUDA, please see the PyTorch installation instructions to see how to specify the desired torch version and --extra-index-url flag. For example, to install with PyTorch 1.12 and CUDA 11.6, use the following command:

$ pip install ml4gw torch==1.12.0 --extra-index-url=https://download.pytorch.org/whl/cu116