Implementation of reconstruction operators in CUDA for Tensorflow for the PYRO-NN Framework
Project description
pyronn-layers
This is a Python wrapper around PYRO-NN-Layers. Install it via
pip install pyronn-layers
Or if you downloaded this repository
pip install -e .
It uses pystencils-autodiff to automatically compile the C++/CUDA source files. The code gets automatically recompiled if you make changes to the source files in the folder PYRO-NN-Layers.
If you prefer pre-build versions of PYRO-NN-Layers go to the Release page of PYRO-NN-Layers or download the latest artifacts from our CI (links only working if pipeline badge above is green):
Requirements
You have a functioning toolchain for NVCC installed. nvcc must be available in PATH. If compilation fails, you can change the selected compiler: http://pycodegen.pages.walberla.net/pystencils/sphinx/configuration.html?highlight=compiler
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.