Skip to content

bonanyan/pyjuice

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

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