Lightweight CUDA tensor functions with a focus on image data augmentation.
Project description
augpy is a lightweight library with minimal dependencies that provides CUDA tensor functions with a focus on image data augmentation. It uses the dlpack standard to export tensors to other libraries, such as Pytorch, with zero copying.
Building
Make sure you have the following installed.
Compiler with C++14 support (e.g. GCC 5)
Cuda 10 or higher
Cmake 3.13 or higher
setuptools>=44.0.0
wheel>=0.34.0
pybind11>=2.4.3
numpy>=1.15.0
Now simply run setup.py as normal to build the wheel and install with pip.
python setup.py bdist_wheel
Usage
TODO
Changelog
1.0.0a1
WIP
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.
Source Distribution
augpy-1.0.0a1.tar.gz
(440.4 kB
view details)
File details
Details for the file augpy-1.0.0a1.tar.gz.
File metadata
- Download URL: augpy-1.0.0a1.tar.gz
- Upload date:
- Size: 440.4 kB
- Tags: Source
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/3.1.1 pkginfo/1.5.0.1 requests/2.21.0 setuptools/40.6.3 requests-toolbelt/0.9.1 tqdm/4.39.0 CPython/3.6.10
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 | d8e96881b53c782f2ee3cba86c11b0c23f0da975235c94f8b5c1ec44a14cf7ef |
|
| MD5 | 5f977e2b1fa89f41e3d54f896526d415 |
|
| BLAKE2b-256 | 485070a6708c87c3db7c3d3e1fb6959445c607d742a33d9197aa0283aac98b5e |
