Skip to content

msaremi/sn2

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

25 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Structural System Neural Network (SN²) Solver CUDA Torch Extension

Reference

Codes of the article Neural Network Parameter-optimization of Gaussian pmDAGs

Installation

Installation from pre-compiled .egg

Pre-compiled Python 3.9 and Python 3.12 Windows eggs are available here. You can install them using the following commands:

cd sn2/dist
wheel convert <egg file name>
pip3 install <created .whl file>

In place of <egg file name> the name of the .egg file should be written. It creates a .whl file, which is used in place of <created .whl file>.

Installation from source

Installation requires the NVCC compiler with an ABI-compatible C++ compiler. On Windows, you will probably need to install MSVC++ and the CUDA toolkit in addition to a GPU-enabled PyTorch package.

To install from source, first move to the package directory. Then, use the pip3 command to install it:

cd sn2
pip3 install .

Alternatively, you may run python setup.py install to manually install the package.

Examples

Examples:

Find more examples here.

About

Parameterizing pmDAGs

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

 
 
 

Contributors