/pyjuice

Primary LanguagePythonApache License 2.0Apache-2.0

pyjuice

CUDA CI Tests codecov

Probabilistic Circuits Package

Installation

  1. (Optional) Make a new conda environment

    conda create -n pyjuice python=3.8
conda activate pyjuice

  2. Clone this repository and cd into it.

  3. Install the pyjuice package in developement mode, run the following:

    pip install --editable .

  4. Install GPU Enabled Pytorch 2.0. See pytorch installation guide for more details. Make sure to install version >=2.0.

    pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118

Testing

  • Install pytest:

    pip install pytest

  • To run the tests, simply call:

    pytest