glearn
is a modular and high-performance Python package for machine learning using Gaussian process regression with novel algorithms capable of petascale computation on multi-GPU devices.
pip install glearn
conda install -c s-ameli glearn
docker pull sameli/glearn
Successful installation and tests performed on the following operating systems, architectures, and Python and PyPy versions:
Python wheels for glearn
for all supported platforms and versions in the above are available through PyPI and Anaconda Cloud. If you need glearn
on other platforms, architectures, and Python or PyPy versions, raise an issue on GitHub and we build its Python Wheel for you.
glearn
can run on CUDA-capable multi-GPU devices. Using the docker container is the easiest way to run glearn
on GPU devices. The supported GPU micro-architectures and CUDA version are as follows:
Version \ Arch | Fermi | Kepler | Maxwell | Pascal | Volta | Turing | Ampere | Hopper |
---|---|---|---|---|---|---|---|---|
CUDA 9 | ✖ | ✖ | ✖ | ✖ | ✖ | ✖ | ✖ | ✖ |
CUDA 10 | ✖ | ✔ | ✔ | ✔ | ✔ | ✔ | ✔ | ✔ |
CUDA 11 | ✖ | ✖ | ✖ | ✔ | ✔ | ✔ | ✔ | ✔ |
CUDA 12 | ✖ | ✖ | ✖ | ✔ | ✔ | ✔ | ✔ | ✔ |
See documentation, including:
- What This Packages Does?
- Comprehensive Installation Guide
- How to Work with Docker Container?
- How to Deploy on GPU Devices?
- API Reference
- Interactive Notebook Tutorials
- Publications
We welcome contributions via GitHub's pull request. If you do not feel comfortable modifying the code, we also welcome feature requests and bug reports as GitHub issues.
If you publish work that uses glearn
, please consider citing the manuscripts available here.
This project uses a BSD 3-clause license, in hopes that it will be accessible to most projects. If you require a different license, please raise an issue and we will consider a dual license.