Nebulae
A deep learning framework based on PyTorch and prevalent image processing libraries
It aims to offer a set of useful tools and functions.
🚀 Spotlight
Several frequently used features are integrated with simple interfaces.
🛠️ Integrated Feats - EMA module, multi-source datasets, training log plotting and timer etc.
🎯 Simplified API - Unified API for distributed and single-GPU training.
⚡️ Efficiency - Data augmentations are reimplemented using Numpy which is faster than PIL.
🧩 High Compatibility - Users are able to build networks using Nebulae with PyTorch seamlessly.
⚡ Quick Start
📸 Utility
Obtain GPU stats after some training epochs.
import nebulae as neb
from nebulae import *
gu = kit.GPUtil()
gu.monitor()
for epoch in range(10):
# --- training code --- #
gu.status()
Automatically select unoccupied GPUs. It is useful for a shared machine.
import nebulae as neb
from nebulae import *
# select 4 GPUs with 2GB or more memory left
engine = npe(device=power.GPU, ngpu=4, least_mem=2048)
Find entire distributed training and test code in ./examples/demo_core.py
📦 Installation
Users can install nebulae from pip
pip install nebulae
For better development, building from Dockerfile is also available. Modifying the libs version and have nvidia-docker on your machine is recommended.
sudo docker build -t nebulae:std -f Dockerfile.std .
sudo docker run -it --gpus all --ipc=host --ulimit memlock=-1 nebulae:std
The latest version supports PyTorch1.6 and above
❤️ Support
If you find Nebulae helpful, consider giving it a ⭐ on GitHub! ▶️ https://github.com/SeriaQ/Nebulae
1 maintainer
Metadata
1 maintainer
Release files for nebulae 0.6.28
For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.
Source distribution (sdist)
| File | Size | Uploaded | |
|---|---|---|---|
| nebulae-0.6.28.tar.gz | 59.0 kB | Details |
Built distribution (wheel)
| File | Interpreter | ABI | Platform | Reset |
|---|---|---|---|---|
| nebulae-0.6.28-py3-none-any.whl | Python 3 | none | any | Details |
Total release size: 123.3 kB
Release files / nebulae-0.6.28.tar.gz
| Download URL | nebulae-0.6.28.tar.gz |
|---|---|
| Size | 59.0 kB |
| Tags | Source |
|
SHA-256 checksum How to use checksums |
b727e1f216ec47691aff487eb8ada2473598aac0f7f94c83855db7a9aed8dec6
|
|
BLAKE2b-256 checksum How to use checksums |
d88c930c4ae63b61ae52b3e62257020cd5587409e5e669e4735f4985261ed072
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
twine/6.1.0 CPython/3.10.12
|
Release files / nebulae-0.6.28-py3-none-any.whl
| Download URL | nebulae-0.6.28-py3-none-any.whl |
|---|---|
| Size | 64.3 kB |
| Tags | Python 3 |
|
SHA-256 checksum How to use checksums |
1ee00b163cad688094dcf44c94563c2b27e2644c4bc392197b5055f4ef184e35
|
|
BLAKE2b-256 checksum How to use checksums |
4692983203be91460f81f724f5448aff65b3229e1a5b24b200ff23f25aed0337
|
| Upload date | |
|
Uploaded using Trusted Publishing? What is trusted publishing? |
No |
| Uploaded via |
twine/6.1.0 CPython/3.10.12
|
