Skip to content

april-tools/pyjuice

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

84 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

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